thinking

app.py

@@ -1,3 +1,182 @@
+# # import gradio as gr
+# # from openai import OpenAI
+
+# # # 🔹 Configure your agent
+# # agent_endpoint = "https://q77iuwf7ncfemoonbzon2iyd.agents.do-ai.run/api/v1/"
+# # agent_access_key = "CzIwmTIDFNWRRIHvxVNzKWztq8rn5S5w"
+
+# # client = OpenAI(base_url=agent_endpoint, api_key=agent_access_key)
+
+# # # Parameters
+# # DEFAULT_RETRIEVAL_RUNS = 3  # adjustable in UI
+
+# # def policy_chat(message, history, retrieval_runs=DEFAULT_RETRIEVAL_RUNS):
+# #     """
+# #     Chatbot with streaming + multiple retrieval runs.
+# #     """
+# #     # Show "processing" placeholder
+# #     history = history + [[message, "Processing..."]]
+# #     yield history, history  
+
+# #     aggregated_responses = []
+
+# #     for run in range(retrieval_runs):
+# #         try:
+# #             stream = client.chat.completions.create(
+# #                 model="n/a",  # agent handles routing
+# #                 messages=[
+# #                     {"role": "system", "content": "The data must be returned verbatim. Please be quite detailed and include all information. You are new to the analysis of policy documents, hence you need to be objective in retrieving information, and it is not expected that you will analyse and interpret the information."},
+# #                     *[{"role": "user", "content": u} if i % 2 == 0 else {"role": "assistant", "content": b}
+# #                       for i, (u, b) in enumerate(history[:-1])],  # exclude placeholder
+# #                     {"role": "user", "content": message},
+# #                 ],
+# #                 extra_body={"include_retrieval_info": True},
+# #                 stream=True,
+# #             )
+
+# #             response_text = ""
+# #             for chunk in stream:
+# #                 delta = chunk.choices[0].delta
+# #                 if delta and delta.content:  # delta.content is a string or None
+# #                     response_text += delta.content
+# #                     # Stream update to Gradio UI
+# #                     history[-1][1] = response_text
+# #                     yield history, history
+
+# #             aggregated_responses.append(response_text or "⚠️ Empty response")
+
+# #         except Exception as e:
+# #             aggregated_responses.append(f"⚠️ Error during run {run+1}: {str(e)}")
+
+# #     # 🔹 Choose the “best” response (longest for now)
+# #     best_response = max(aggregated_responses, key=len, default="⚠️ No response")
+# #     history[-1][1] = best_response
+# #     yield history, history
+
+
+# # with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray")) as demo:
+# #     gr.Markdown("# 🤖 Policy-Agent Chatbot\nAsk me about policies. I’ll query the knowledge base multiple times to retrieve the best answer for you!")
+
+# #     popup_widget_html = '''<style>
+# #         .policy-agent-popup-container { position:fixed; bottom:16px; right:16px; z-index:9999; }
+# #         </style>
+# #         <div class="policy-agent-popup-container">
+# #             <script async
+# #                 src="https://q77iuwf7ncfemoonbzon2iyd.agents.do-ai.run/static/chatbot/widget.js"
+# #                 data-agent-id="fcad9141-8590-11f0-b074-4e013e2ddde4"
+# #                 data-chatbot-id="oTQKgtWMkQLbLVw7CIHkbxw25Pu9jekn"
+# #                 data-name="policy-agent Chatbot"
+# #                 data-primary-color="#031B4E"
+# #                 data-secondary-color="#E5E8ED"
+# #                 data-button-background-color="#0061EB"
+# #                 data-starting-message="Hello! I am your policy analysis bot made to help you comb through the policies."
+# #                 data-logo="/static/chatbot/icons/default-agent.svg">
+# #             </script>
+# #         </div>'''
+# #     gr.HTML(popup_widget_html)
+
+# #     chatbot = gr.Chatbot(height=500)
+# #     msg = gr.Textbox(placeholder="Type your question...")
+# #     retrieval_slider = gr.Slider(1, 10, value=DEFAULT_RETRIEVAL_RUNS, step=1, label="Number of retrieval runs")
+# #     clear = gr.Button("Clear Chat")
+
+# #     msg.submit(policy_chat, [msg, chatbot, retrieval_slider], [chatbot, chatbot])
+# #     msg.submit(lambda: "", None, msg)  # clear textbox
+# #     clear.click(lambda: None, None, chatbot, queue=False)
+
+# # if __name__ == "__main__":
+# #     demo.launch(debug=True)
+
+# import gradio as gr
+# from openai import OpenAI
+
+# # 🔹 Configure your agent
+# agent_endpoint = "https://q77iuwf7ncfemoonbzon2iyd.agents.do-ai.run/api/v1/"
+# agent_access_key = "CzIwmTIDFNWRRIHvxVNzKWztq8rn5S5w"
+
+# client = OpenAI(base_url=agent_endpoint, api_key=agent_access_key)
+
+# # Parameters
+# DEFAULT_RETRIEVAL_RUNS = 3  # adjustable in UI
+# stop_flag = {"stop": False}  # global flag for stopping runs
+
+# def stop_runs():
+#     """Set stop flag to True to interrupt current processing."""
+#     stop_flag["stop"] = True
+#     return "⛔ Retrieval stopped by user."
+
+
+# def policy_chat(message, history, retrieval_runs=DEFAULT_RETRIEVAL_RUNS):
+#     """
+#     Runs multiple retrievals, stores responses, and returns the best one.
+#     """
+#     # Reset stop flag
+#     stop_flag["stop"] = False
+
+#     # Show "processing" placeholder in UI
+#     history = history + [[message, "Processing..."]]
+#     yield history, history  
+
+#     aggregated_responses = []
+
+#     for run in range(retrieval_runs):
+#         if stop_flag["stop"]:
+#             history[-1][1] = "⛔ Stopped before completion."
+#             yield history, history
+#             return
+
+#         try:
+#             # Stream response but collect full text silently
+#             stream = client.chat.completions.create(
+#                 model="n/a",  # agent handles routing
+#                 messages=[
+#                     {"role": "system", "content": "The data must be returned verbatim. Please be quite detailed and include all information. You are new to the analysis of policy documents, hence you need to be objective in retrieving information, and it is not expected that you will analyse and interpret the information."},
+#                     *[{"role": "user", "content": u} if i % 2 == 0 else {"role": "assistant", "content": b}
+#                       for i, (u, b) in enumerate(history[:-1])],  # exclude placeholder
+#                     {"role": "user", "content": message},
+#                 ],
+#                 extra_body={"include_retrieval_info": True},
+#                 stream=True,
+#             )
+
+#             response_text = ""
+#             for chunk in stream:
+#                 delta = chunk.choices[0].delta
+#                 if delta and delta.content:
+#                     response_text += delta.content
+
+#             aggregated_responses.append(response_text or "⚠️ Empty response")
+
+#         except Exception as e:
+#             aggregated_responses.append(f"⚠️ Error during run {run+1}: {str(e)}")
+
+#     # --- Selection logic ---
+#     # For now, pick the "best" as the longest response
+#     best_response = max(aggregated_responses, key=len, default="⚠️ No response")
+
+#     # Replace "Processing..." with final answer
+#     history[-1][1] = best_response
+#     yield history, history
+
+
+# with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray")) as demo:
+#     gr.Markdown("# 🤖 Policy-Agent Chatbot\nAsk me about policies. I’ll query the knowledge base multiple times!")
+
+#     chatbot = gr.Chatbot(height=500)
+#     msg = gr.Textbox(placeholder="Type your question...")
+#     retrieval_slider = gr.Slider(1, 10, value=DEFAULT_RETRIEVAL_RUNS, step=1, label="Number of retrieval runs")
+#     with gr.Row():
+#         clear = gr.Button("Clear Chat")
+#         stop = gr.Button("⛔ Stop")
+
+#     msg.submit(policy_chat, [msg, chatbot, retrieval_slider], [chatbot, chatbot])
+#     msg.submit(lambda: "", None, msg)  # clear textbox
+#     clear.click(lambda: None, None, chatbot, queue=False)
+#     stop.click(stop_runs, None, chatbot, queue=False)
+
+# if __name__ == "__main__":
+#     demo.launch(debug=True)
+
 import gradio as gr
 from openai import OpenAI
 from datetime import datetime
@@ -62,7 +241,13 @@ def policy_chat(message, history):
 
     # Add user message and empty assistant response to history
     history = history + [[message, ""]]
-    yield {chatbot: history, thinking_indicator: gr.update(visible=True)}
+    yield {
+        chatbot: history, 
+        thinking_indicator: gr.update(visible=True),
+        reasoning_accordion: gr.update(visible=False, value=""),
+        stop_btn: gr.update(visible=True),
+        send_btn: gr.update(interactive=False)
+    }
 
     try:
         # 2. Simulate reasoning (this would be your retrieval logic integrated with DigitalOcean agent)
@@ -78,7 +263,7 @@ def policy_chat(message, history):
         yield {
             chatbot: history,
             thinking_indicator: gr.update(visible=False),
-            reasoning_accordion: gr.update(visible=False, value=REASONING_CONTEXT),
+            reasoning_accordion: gr.update(visible=True, value=REASONING_CONTEXT),
             stop_btn: gr.update(visible=True),
             send_btn: gr.update(interactive=False)
         }
@@ -103,7 +288,13 @@ def policy_chat(message, history):
             if stop_flag["stop"]:
                 response_text += "\n\n⛔ **Streaming stopped by user.**"
                 history[-1][1] = response_text
-                yield {chatbot: history}
+                yield {
+                    chatbot: history,
+                    thinking_indicator: gr.update(visible=False),
+                    reasoning_accordion: gr.update(visible=True, value=REASONING_CONTEXT),
+                    stop_btn: gr.update(visible=True),
+                    send_btn: gr.update(interactive=False)
+                }
                 break  # Exit the loop
 
             delta = chunk.choices[0].delta
@@ -111,19 +302,33 @@ def policy_chat(message, history):
                 response_text += delta.content
                 # Update the last message in history with streaming content
                 history[-1][1] = response_text
-                yield {chatbot: history}
+                yield {
+                    chatbot: history,
+                    thinking_indicator: gr.update(visible=False),
+                    reasoning_accordion: gr.update(visible=True, value=REASONING_CONTEXT),
+                    stop_btn: gr.update(visible=True),
+                    send_btn: gr.update(interactive=False)
+                }
 
     except Exception as e:
         # Handle errors by updating the assistant message
         history[-1][1] = f"⚠️ Error: {str(e)}"
-        yield {chatbot: history}
+        yield {
+            chatbot: history,
+            thinking_indicator: gr.update(visible=False),
+            reasoning_accordion: gr.update(visible=True, value=REASONING_CONTEXT),
+            stop_btn: gr.update(visible=False),
+            send_btn: gr.update(interactive=True)
+        }
     
     finally:
         # Hide the stop button and re-enable send button once streaming is complete or stopped
         yield {
+            chatbot: history,
+            thinking_indicator: gr.update(visible=False),
+            reasoning_accordion: gr.update(visible=True, value=REASONING_CONTEXT),
             stop_btn: gr.update(visible=False),
-            send_btn: gr.update(interactive=True),
-            thinking_indicator: gr.update(visible=False)
+            send_btn: gr.update(interactive=True)
         }