Updated Script for API Key

app.py

@@ -19,7 +19,47 @@ 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
+    return ChatOpenAI(temperature=0)
+
+def process_query(api_key: str, query: str):
+    try:
+        llm = init_llm(api_key)
+        agent = Agent(
+            task=query,
+            llm=llm
+        )
+        result = asyncio.run(agent.run())
+        return result
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Customer Support Agent")
+    with gr.Row():
+        api_key = gr.Textbox(
+            label="OpenAI API Key", 
+            placeholder="Enter your OpenAI API key",
+            type="password"
+        )
+    with gr.Row():
+        query = gr.Textbox(
+            label="Query",
+            placeholder="Enter your query here"
+        )
+    with gr.Row():
+        submit = gr.Button("Submit")
+    with gr.Row():
+        output = gr.Textbox(label="Response")
+    
+    submit.click(
+        fn=process_query,
+        inputs=[api_key, query],
+        outputs=output
+    )
 # Initialize our language models.
 # We use llm_standard for normal tasks and llm_browser for browser-based tasks.
 llm_standard = ChatOpenAI(temperature=0)