Update app.py

app.py

@@ -71,20 +71,20 @@ from langchain_openai import ChatOpenAI
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
 
-model_options = ["gpt-4o", "gpt-4o-mini"]
-selected_model = st.selectbox("Choose a GPT model", model_options)
-
-llm = ChatOpenAI(
-    model=selected_model,#"gpt-4o-mini",
-    temperature=0,
-    max_tokens=None,
-    timeout=None,
-    max_retries=2,
-    api_key=os.getenv("OPENAI_API_KEY"),  # if you prefer to pass api key in directly instaed of using env vars
-    # base_url="...",
-    # organization="...",
-    # other params...
-)
+# model_options = ["gpt-4o", "gpt-4o-mini"]
+# selected_model = st.selectbox("Choose a GPT model", model_options)
+
+# llm = ChatOpenAI(
+#     model=selected_model,#"gpt-4o-mini",
+#     temperature=0,
+#     max_tokens=None,
+#     timeout=None,
+#     max_retries=2,
+#     api_key=os.getenv("OPENAI_API_KEY"),  # if you prefer to pass api key in directly instaed of using env vars
+#     # base_url="...",
+#     # organization="...",
+#     # other params...
+# )
 
 # default_system_prompt = """
 # You are a helpful and knowledgeable assistant who is expert on medical question answering. 
@@ -267,6 +267,7 @@ def main():
         page_title="Medical Chatbot",
         page_icon=favicon_image,
     )
+    
     # Create two columns for the logo and title text
     col1, col2 = st.columns([1, 8])  # Adjust the column width ratios as needed
     
@@ -316,6 +317,20 @@ def main():
 
 
     user_question = st.chat_input("Ask a Question")
+    model_options = ["gpt-4o", "gpt-4o-mini"]
+    selected_model = st.selectbox("Choose a GPT model", model_options)
+    
+    llm = ChatOpenAI(
+        model=selected_model,#"gpt-4o-mini",
+        temperature=0,
+        max_tokens=None,
+        timeout=None,
+        max_retries=2,
+        api_key=os.getenv("OPENAI_API_KEY"),  # if you prefer to pass api key in directly instaed of using env vars
+        # base_url="...",
+        # organization="...",
+        # other params...
+    )
 
     # Initialize or load chat history into session state
     if "messages" not in st.session_state: