add api key input option

app.py

@@ -14,12 +14,12 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 if TAVILY_API_KEY:
     print("TAVILY_API_KEY found!")
 else:
-    print("TAVILY_API_KEY not found. Please check your Secrets configuration.")
+    print("Please provide your TAVILY_API_KEY.")
 
 if OPENAI_API_KEY:
     print("OPENAI_API_KEY found!")
 else:
-    print("OPENAI_API_KEY not found. Please check your Secrets configuration.")
+    print("Please provide your OPENAI_API_KEY not found")
 
 def extract_leaf_nodes(data, parent_key=''):
     leaf_nodes = {}
@@ -77,6 +77,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### 🛠 Input")
+            tavily_key_input = gr.Textbox(label="Tavily API Key", type="password",
+                                          placeholder="Enter your Tavily API Key")
+            openai_key_input = gr.Textbox(label="OpenAI API Key", type="password",
+                                          placeholder="Enter your OpenAI API Key")
             schema_input = gr.Textbox(
                 label="Extraction Schema (JSON)",
                 value=json.dumps({
@@ -99,10 +103,22 @@ with gr.Blocks() as demo:
             output_display = gr.Markdown(label="Extracted Information")
             time_display = gr.Textbox(label="Processing Time (seconds)", interactive=False)
 
-    def on_submit(schema, topic):
-        data, time_taken = agent_response(schema, topic)
-        return data, f"{time_taken:.2f}"
 
-    submit_button.click(on_submit, inputs=[schema_input, topic_input], outputs=[output_display, time_display])
+    def on_submit(schema, topic, tavily_key, openai_key):
+        try:
+            # Set API keys as environment variables
+            os.environ["TAVILY_API_KEY"] = tavily_key
+            os.environ["OPENAI_API_KEY"] = openai_key
+
+            # Call agent_response
+            data, time_taken = agent_response(schema, topic)
+            return data, f"{time_taken:.2f}"
+        except Exception as e:
+            return f"❌ An error occurred: {str(e)}", "0.00"
+
+
+    submit_button.click(on_submit,
+                        inputs=[schema_input, topic_input, tavily_key_input, openai_key_input],
+                        outputs=[output_display, time_display])
 
 demo.launch(share=True)