Update app.py

app.py

@@ -317,10 +317,36 @@ def main():
 
 
     user_question = st.chat_input("Ask a Question")
-    model_options = ["gpt-4o", "gpt-4o-mini"]
+    model_options = ["gpt-4o", "gpt-4o-mini","deepseek-chat"]
     selected_model = st.selectbox("Choose a GPT model", model_options)
+
+    if selected_model == "deepseek-chat":
     
-    # llm = ChatOpenAI(
+        llm = ChatOpenAI(
+            model=selected_model,#"gpt-4o-mini",
+            temperature=0,
+            max_tokens=None,
+            timeout=None,
+            max_retries=2,
+            api_key=os.getenv("DeepSeek_API_KEY"),  # if you prefer to pass api key in directly instaed of using env vars
+            base_url="https://api.deepseek.com",
+            # organization="...",
+            # other params...
+        )
+    else:
+        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...
+        )     
+        
+    # llm = OpenAI(
     #     model=selected_model,#"gpt-4o-mini",
     #     temperature=0,
     #     max_tokens=None,
@@ -331,17 +357,6 @@ def main():
     #     # organization="...",
     #     # other params...
     # )
-    llm = OpenAI(
-        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: