Update app.py

app.py

@@ -23,46 +23,62 @@ import hashlib
 
 load_dotenv()
 
-def check_password():
-    def password_entered():
-        if hmac.compare_digest(st.session_state["username"], os.environ.get("STREAMLIT_USERNAME", "")) and \
-           hmac.compare_digest(st.session_state["password"], os.environ.get("STREAMLIT_PASSWORD", "")):
-            st.session_state["password_correct"] = True
-            del st.session_state["password"]
-            del st.session_state["username"]
+def setup_api_clients():
+    with st.sidebar:
+        st.title("API Configuration")
+        
+        use_stored = st.checkbox("Use stored API keys")
+        
+        if use_stored:
+            username = st.text_input("Username")
+            password = st.text_input("Password", type="password")
+            
+            if st.button("Verify Credentials"):
+                if (hmac.compare_digest(username, os.environ.get("STREAMLIT_USERNAME", "")) and 
+                    hmac.compare_digest(password, os.environ.get("STREAMLIT_PASSWORD", ""))):
+                    st.session_state.togetherai_client = OpenAI(
+                        api_key=os.getenv('TOGETHERAI_API_KEY'),
+                        base_url="https://api.together.xyz/v1"
+                    )
+                    st.session_state.openai_client = OpenAI(
+                        api_key=os.getenv('OPENAI_API_KEY')
+                    )
+                    st.session_state.anthropic_client = Anthropic(
+                        api_key=os.getenv('ANTHROPIC_API_KEY')
+                    )
+                    genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+                    
+                    st.session_state.api_configured = True
+                    st.success("Successfully configured API clients with stored keys!")
+                else:
+                    st.error("Invalid credentials. Please try again or use your own API keys.")
+                    st.session_state.api_configured = False
         else:
-            st.session_state["password_correct"] = False
+            st.subheader("Enter Your API Keys")
+            togetherai_key = st.text_input("Together AI API Key", type="password", key="togetherai_key")
+            openai_key = st.text_input("OpenAI API Key", type="password", key="openai_key")
+            anthropic_key = st.text_input("Anthropic API Key", type="password", key="anthropic_key")
+            gemini_key = st.text_input("Gemini API Key", type="password", key="gemini_key")
             
-    if "password_correct" not in st.session_state:
-        st.text_input("Username", key="username")
-        st.text_input("Password", type="password", key="password")
-        st.button("Login", on_click=password_entered)
-        return False
-    
-    # Password correct
-    elif not st.session_state["password_correct"]:
-        st.text_input("Username", key="username")
-        st.text_input("Password", type="password", key="password")
-        st.button("Login", on_click=password_entered)
-        st.error("Wrong Credentials Entered")
-        return False
-    
-    return True
-
-togetherai_client = OpenAI(
-    api_key=os.getenv('TOGETHERAI_API_KEY'),
-    base_url="https://api.together.xyz/v1"
-)
-
-openai_client = OpenAI(
-    api_key=os.getenv('OPENAI_API_KEY')
-)
-
-anthropic_client = Anthropic(
-    api_key=os.getenv('ANTHROPIC_API_KEY')
-)
-
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+            if st.button("Initialize with provided keys"):
+                try:
+                    st.session_state.togetherai_client = OpenAI(
+                        api_key=togetherai_key,
+                        base_url="https://api.together.xyz/v1"
+                    )
+                    st.session_state.openai_client = OpenAI(
+                        api_key=openai_key
+                    )
+                    st.session_state.anthropic_client = Anthropic(
+                        api_key=anthropic_key
+                    )
+                    genai.configure(api_key=gemini_key)
+                    
+                    st.session_state.api_configured = True
+                    st.success("Successfully configured API clients with provided keys!")
+                except Exception as e:
+                    st.error(f"Error initializing API clients: {str(e)}")
+                    st.session_state.api_configured = False
 
 MAX_CONCURRENT_CALLS = 5
 semaphore = threading.Semaphore(MAX_CONCURRENT_CALLS)
@@ -202,13 +218,21 @@ def process_evaluations_concurrently(questions, prompt_template, models_to_evalu
             progress_callback(current_iteration, total_iterations)
     
     return results
-
+    
 def main():
     st.set_page_config(page_title="LLM Benchmarking in Healthcare", layout="wide")
     
-    if not check_password():
-        st.stop()
+    if 'api_configured' not in st.session_state:
+        st.session_state.api_configured = False
+    
+    setup_api_clients()
+    
     st.title("LLM Benchmarking in Healthcare")
+    
+    if not st.session_state.api_configured:
+        st.warning("Please configure API keys in the sidebar to proceed")
+        st.stop()
+    
 
     if 'all_results' not in st.session_state:
         st.session_state.all_results = {}