Update app.py

app.py

@@ -1,8 +1,16 @@
 import gradio as gr
 from agent import start_scraping, display_csv, generate_rss_feed, chat_interface
 
-# Create Gradio interface
+def validate_input(values):
+    """Validate user input"""
+    if not values["storage_location"]:
+        raise ValueError("Storage location cannot be empty")
+    if not values["urls"]:
+        raise ValueError("URLs cannot be empty")
+    return values
+
 def create_interface():
+    """Create Gradio interface"""
     with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
@@ -22,8 +30,25 @@ def create_interface():
                 chat_history = gr.Chatbot(label="Chat History")
                 response_box = gr.Textbox(label="Response")
 
-        start_button.click(start_scraping, inputs=[storage_location, urls, scrape_interval, content_type], outputs=csv_output)
-        message.submit(chat_interface, inputs=[message, chat_history, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type], outputs=[chat_history, response_box])
+        def start_scraping_callback(values):
+            """Start scraping callback"""
+            try:
+                validated_values = validate_input(values)
+                start_scraping(validated_values["storage_location"], validated_values["urls"], validated_values["scrape_interval"], validated_values["content_type"])
+                display_csv(validated_values["storage_location"])
+            except Exception as e:
+                print(f"Error: {str(e)}")
+
+        start_button.click(start_scraping_callback, inputs=[storage_location, urls, scrape_interval, content_type], outputs=csv_output)
+
+        def chat_interface_callback(values):
+            """Chat interface callback"""
+            try:
+                chat_interface(values["message"], chat_history, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type)
+            except Exception as e:
+                print(f"Error: {str(e)}")
+
+        message.submit(chat_interface_callback, inputs=[message, chat_history, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type], outputs=[chat_history, response_box])
 
         # Add a button to display the RSS feed for a selected URL
         with gr.Row():
@@ -31,7 +56,14 @@ def create_interface():
             rss_button = gr.Button("Generate RSS Feed")
             rss_output = gr.Textbox(label="RSS Feed Output", interactive=False)
 
-        rss_button.click(generate_rss_feed, inputs=[selected_url], outputs=rss_output)
+        def generate_rss_feed_callback(values):
+            """Generate RSS feed callback"""
+            try:
+                generate_rss_feed(values["selected_url"])
+            except Exception as e:
+                print(f"Error: {str(e)}")
+
+        rss_button.click(generate_rss_feed_callback, inputs=[selected_url], outputs=rss_output)
 
     return demo