openrouter

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from openai import OpenAI
 from together import Together
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
 from datetime import datetime
 import time
 
@@ -51,7 +53,7 @@ if 'inserted' not in st.session_state:
     st.session_state.gotit = False
     st.session_state.submitted = False
     st.session_state.inserted = 0
-    st.session_state["model"] = "gpt-4o-2024-08-06" # "o3-mini-2025-01-31" #"deepseek-ai/DeepSeek-V3"
+    st.session_state["model"] = "deepseek/deepseek-chat" #"deepseek/deepseek-r1" #"gpt-4o-2024-08-06" # "o3-mini-2025-01-31" #"deepseek-ai/DeepSeek-V3"
     st.session_state.max_messages = 50
     st.session_state.messages = []
     st.session_state.user_data = {}
@@ -89,7 +91,8 @@ def setup_messages():
     st.session_state.convo_start_time = datetime.now()
 
 if st.session_state.model.startswith("deepseek"):
-    client = Together(api_key=st.secrets["TOGETHER_API_KEY"])
+    client = OpenAI( base_url="https://openrouter.ai/api/v1", api_key=st.secrets["OPENROUTER_API_KEY"])
+    #client = Together(api_key=st.secrets["TOGETHER_API_KEY"])
 else:
     client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
 
@@ -158,7 +161,7 @@ def form():
     columns_form = st.columns((1,1,1))
     with columns_form[2]:
         submitted = st.button("Proceed",use_container_width=True, 
-                              help = 'Please answer every question and click *Proceed* to start a conversation.',
+                              help = 'Make sure you answer every question.',
                               disabled = not (all(st.session_state[field] != '' for field in st.session_state.fields) and st.session_state.recycling != 0))
 
     if submitted:
@@ -170,8 +173,6 @@ def form():
         st.session_state.user_data["inserted"] = st.session_state.inserted
         st.session_state.user_data["submission_time"] = datetime.now()
         
-        from pymongo.mongo_client import MongoClient
-        from pymongo.server_api import ServerApi
         with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
                 db = client.chat
                 collection = db.app
@@ -212,10 +213,6 @@ def submit():
         
         st.session_state.user_data["convo_end_time"] = datetime.now()
 
-        st.write(st.session_state.user_data)
-
-        from pymongo.mongo_client import MongoClient
-        from pymongo.server_api import ServerApi
         with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
                 db = client.chat
                 collection = db.app
@@ -268,6 +265,7 @@ elif prompt := st.chat_input("Ask a question about climate action..."):
             )
             
             if st.session_state.model.startswith("deepseek"):
+                
                 response = st.write_stream(generate_tokens(stream)) 
             else:
                 response = st.write_stream(stream)