Update app.py

app.py

@@ -25,6 +25,32 @@ assert os.getenv("GOOGLE_API_KEY"), "Set GOOGLE_API_KEY in .env"
 assert os.getenv("TAVILY_API_KEY"), "Set TAVILY_API_KEY in .env"
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
+
+service_account_path = {
+  "type": os.getenv("type"),
+  "project_id": os.getenv("project_id"),
+  "private_key_id": str(os.getenv("private_key_id")).replace('\\n', '\n'),
+  "private_key":  str(os.getenv("private_key")).replace('\\n', '\n'),
+  "client_email": os.getenv("client_email"),
+  "client_id": os.getenv("client_id"),
+  "auth_uri": os.getenv("auth_uri"),
+  "token_uri": os.getenv("token_uri"),
+  "auth_provider_x509_cert_url": os.getenv("auth_provider_x509_cert_url"),
+  "client_x509_cert_url": os.getenv("client_x509_cert_url"),
+  "universe_domain":os.getenv("universe_domain")
+}
+
+def add_query_to_sheet(user_id, query, response):
+    gc = gspread.service_account_from_dict(service_account_path)
+    sh = gc.open_by_key(SHEET_KEY)
+    worksheet = sh.worksheet("Sheet1")
+    
+     
+    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    worksheet.append_row([user_id, timestamp, query, response])
+
+
+
 def flatten_json(obj: dict) -> str:
     pieces = []
     def recurse(prefix, value):
@@ -125,6 +151,15 @@ runner, session = create_agent()
 # === STREAMLIT UI ===
 st.title("🎓 Jossa-ChatBot")
 
+if "user_id" not in st.session_state:
+    st.session_state.user_id = str(uuid.uuid4())   
+if "session_id" not in st.session_state:
+    st.session_state.session_id = str(uuid.uuid4())  
+
+user_id = st.session_state.user_id
+
+runner, session = create_agent(st.session_state.user_id, st.session_state.session_id)
+
 if "history" not in st.session_state:
     st.session_state.history = []
 
@@ -139,10 +174,14 @@ if query:
     st.session_state.history.append(("user", query))
     # Run agent
     user_msg = types.Content(role="user", parts=[types.Part(text=query)])
-    events = runner.run(user_id="user1", session_id=session.id, new_message=user_msg)
+    events = runner.run(user_id=user_id, session_id=session.id, new_message=user_msg)
     # Collect final response text
     reply = ""
     for ev in events:
         if ev.is_final_response(): reply = ev.content.parts[0].text
+    if reply=="":
+        reply="Please provide complete context."
     st.session_state.history.append(("assistant", reply))
+    add_query_to_sheet(user_id=user_id, query=query, response=reply)
     st.rerun()  
+