update: add UID in url and states

app.py

@@ -333,7 +333,7 @@ def get_initial_analysis(
             link_start = "http"
         else:
             link_start = "https"
-
+        print(uid)
         if ct == "si":  # street interview
             prompt = f"""This is a transcript for a street interview. Transcript: {transcript} User ID UID: {uid}
 
@@ -457,6 +457,7 @@ def chat(
     rsid,
     origin,
     ct,
+    uid,
 ) -> str:
     tools = [
         {
@@ -583,7 +584,12 @@ In the URL, make sure that after RSID there is ? and then rest of the fields are
                         if call_type:
                             # Stream the analysis for corrected call type
                             for content in get_initial_analysis(
-                                transcript_processor, call_type, rsid, origin, call_type
+                                transcript_processor,
+                                call_type,
+                                rsid,
+                                origin,
+                                call_type,
+                                uid,
                             ):
                                 yield content
                             return
@@ -738,6 +744,7 @@ def create_chat_interface():
         origin_state = gr.State()
         ct_state = gr.State()
         turl_state = gr.State()
+        uid_state = gr.State()
         iframe_html = "<iframe id='link-frame'></iframe>"
         gr.HTML(value=iframe_html)  # Add iframe to the UI
 
@@ -749,6 +756,7 @@ def create_chat_interface():
             rsid,
             origin,
             ct,
+            uid,
         ):
             if not transcript_processor:
                 bot_message = "Transcript processor not initialized."
@@ -764,6 +772,7 @@ def create_chat_interface():
                 rsid,
                 origin,
                 ct,
+                uid,
             ):
                 chat_history[-1] = (message, chunk)
                 yield "", chat_history
@@ -778,6 +787,7 @@ def create_chat_interface():
                 colab_id_state,
                 origin_state,
                 ct_state,
+                uid_state,
             ],
             [msg, chatbot],
         )
@@ -789,8 +799,10 @@ def create_chat_interface():
             origin = request.query_params.get("origin", None)
             ct = request.query_params.get("ct", None)
             turl = request.query_params.get("turl", None)
+            uid = request.query_params.get("uid", None)
+            print(uid)
 
-            required_params = ["cid", "rsid", "origin", "ct", "turl"]
+            required_params = ["cid", "rsid", "origin", "ct", "turl", "uid"]
             missing_params = [
                 param
                 for param in required_params
@@ -802,15 +814,7 @@ def create_chat_interface():
                     f"Missing required parameters: {', '.join(missing_params)}"
                 )
                 chatbot_value = [(None, error_message)]
-                return [
-                    chatbot_value,
-                    None,
-                    None,
-                    None,
-                    None,
-                    None,
-                    None,
-                ]
+                return [chatbot_value, None, None, None, None, None, None, None]
 
             try:
                 transcript_data = get_transcript_for_url(turl)
@@ -830,26 +834,19 @@ def create_chat_interface():
                     origin,
                     ct,
                     turl,
+                    uid,
                 ]
             except Exception as e:
                 error_message = f"Error processing call_id {cid}: {str(e)}"
                 chatbot_value = [(None, error_message)]
-                return [
-                    chatbot_value,
-                    None,
-                    None,
-                    None,
-                    None,
-                    None,
-                    None,
-                ]
+                return [chatbot_value, None, None, None, None, None, None, None]
 
         def stream_initial_analysis(
-            chatbot_value, transcript_processor, cid, rsid, origin, ct
+            chatbot_value, transcript_processor, cid, rsid, origin, ct, uid
         ):
             if transcript_processor:
                 for chunk in get_initial_analysis(
-                    transcript_processor, cid, rsid, origin, ct
+                    transcript_processor, cid, rsid, origin, ct, uid
                 ):
                     chatbot_value[0] = (None, chunk)
                     yield chatbot_value
@@ -868,6 +865,7 @@ def create_chat_interface():
                 origin_state,
                 ct_state,
                 turl_state,
+                uid_state,
             ],
         ).then(
             stream_initial_analysis,
@@ -878,6 +876,7 @@ def create_chat_interface():
                 colab_id_state,
                 origin_state,
                 ct_state,
+                uid_state,
             ],
             outputs=[chatbot],
         )