Updated APP

app.py

@@ -19,7 +19,6 @@ class State(TypedDict):
     category: str
     sentiment: str
     response: str
-    api_key: str
 
 def init_llm(api_key: str):
     os.environ["OPENAI_API_KEY"] = api_key
@@ -175,21 +174,24 @@ workflow.add_edge("escalate", END)
 workflow.set_entry_point("categorize")
 app = workflow.compile()
 
-def run_customer_support(query: str, api_key: str) -> str:
-    """
-    Process the customer query through the workflow.
-    Use the provided API key, or if none is given, fall back to the .env value.
-    Only the final answer is returned.
-    """
-    # If no API key is provided in the UI, try to read it from the environment.
-    if not api_key.strip():
-        api_key = os.getenv("OPENAI_API_KEY", "")
-        if not api_key:
-            return "Please provide a valid OpenAI API key."
-    os.environ["OPENAI_API_KEY"] = api_key
-    results = app.invoke({"query": query})
-    # Return only the final answer (the response part)
-    return results.get("response", "No response generated.")
+async def run_customer_support(query: str, api_key: str = "") -> str:
+    if not api_key and not os.getenv("OPENAI_API_KEY"):
+        return "Error: Please provide an OpenAI API key"
+    
+    try:
+        if api_key:
+            os.environ["OPENAI_API_KEY"] = api_key
+            
+        llm = ChatOpenAI(temperature=0)
+        agent = Agent(
+            task=query,
+            llm=llm,
+        )
+        result = await agent.run()
+        return result
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 
 # Build the Gradio UI.
 with gr.Blocks(title="Customer Support Agent with Browser Use") as demo:
@@ -198,14 +200,28 @@ with gr.Blocks(title="Customer Support Agent with Browser Use") as demo:
     
     with gr.Row():
         with gr.Column():
-            # The API key textbox (if left empty, the app will try to use the .env key)
-            api_key_input = gr.Textbox(label="OpenAI API Key", type="password", placeholder="sk-...", value="")
-            query_input = gr.Textbox(label="Customer Query", placeholder="Enter your query here...", lines=3)
+            api_key_input = gr.Textbox(
+                label="OpenAI API Key",
+                type="password",
+                placeholder="sk-..."
+            )
+            query_input = gr.Textbox(
+                label="Customer Query",
+                placeholder="Enter your query here...",
+                lines=3
+            )
             submit_btn = gr.Button("Submit Query")
         with gr.Column():
-            output_box = gr.Textbox(label="Agent Response", lines=10, interactive=False)
+            output_box = gr.Textbox(
+                label="Agent Response",
+                lines=10,
+                interactive=False
+            )
     
-    submit_btn.click(fn=run_customer_support, inputs=[query_input, api_key_input], outputs=output_box)
-
+    submit_btn.click(
+        fn=run_customer_support,
+        inputs=[query_input, api_key_input],
+        outputs=output_box
+    )
 if __name__ == "__main__":
     demo.launch()