Update app.py

app.py

@@ -1,74 +1,53 @@
-import gradio as gr
+import os
 from agent import start_scraping, display_csv, generate_rss_feed, chat_interface
+import gradio as gr
 
-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():
-                message = gr.Textbox(label="Message")
-                system_message = gr.Textbox(value="You are a helpful assistant.", label="System message")
-                max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-                temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
-                top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
-                storage_location = gr.Textbox(value="scraped_data", label="Storage Location")
-                urls = gr.Textbox(label="URLs (comma separated)")
-                scrape_interval = gr.Slider(minimum=1, maximum=60, value=5, step=1, label="Scrape Interval (minutes)")
-                content_type = gr.Radio(choices=["text", "media", "both"], value="text", label="Content Type")
-                selector = gr.Textbox(label="CSS Selector (for media, optional)")
-                start_button = gr.Button("Start Scraping")
-                csv_output = gr.Textbox(label="CSV Output", interactive=False)
-                status_output = gr.Textbox(label="Status", interactive=False)
-
-            with gr.Column():
-                chat_history = gr.Chatbot(label="Chat History")
-                response_box = gr.Textbox(label="Response")
-
-        def start_scraping_callback(values):
-            """Start scraping callback"""
-            try:
-                validated_values = validate_input(values)
-                status_output.value = start_scraping(validated_values["storage_location"], validated_values["urls"].split(","), validated_values["scrape_interval"], validated_values["content_type"], validated_values["selector"])
-                csv_output.value = display_csv(validated_values["storage_location"], validated_values["urls"].split(",")[0])  # Display CSV for the first URL
-            except Exception as e:
-                print(f"Error: {str(e)}")
-
-        start_button.click(start_scraping_callback, inputs=[storage_location, urls, scrape_interval, content_type, selector], outputs=[csv_output, status_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.split(","), scrape_interval, content_type, selector)
-            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, selector], outputs=[chat_history, response_box])
-
-        # Add a button to display the RSS feed for a selected URL
-        with gr.Row():
-            selected_url = gr.Textbox(label="Select URL for RSS Feed")
-            rss_button = gr.Button("Generate RSS Feed")
-            rss_output = gr.Textbox(label="RSS Feed Output", interactive=False)
-
-        def generate_rss_feed_callback(values):
-            """Generate RSS feed callback"""
-            try:
-                rss_output.value = generate_rss_feed(storage_location, 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
+def chatbot_interface(message, history, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type, selector):
+    history, response = chat_interface(message, history, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type, selector)
+    return history, response
+
+def generate_rss(storage_location, url):
+    feed_entries = generate_rss_feed(storage_location, url)
+    return feedparser.FeedParserDict(feed_entries)
+
+def main():
+    storage_location = "scraped_data"
+    urls = ["https://www.culvers.com/"]
+    scrape_interval = 5
+    content_type = "text"
+    selector = ""
+
+    chatbot_input = gr.inputs.Textbox(lines=5, label="Chatbot Input")
+    history_output = gr.outputs.Textbox(label="History")
+    response_output = gr.outputs.Textbox(label="Response")
+
+    csv_input = gr.inputs.Textbox(lines=1, label="CSV Input URL")
+    csv_output = gr.outputs.Textbox(label="CSV Output")
+
+    rss_input = gr.inputs.Textbox(lines=1, label="RSS Input URL")
+    rss_output = gr.outputs.HTML(label="RSS Output")
+
+    chatbot_interface = gr.Interface(
+        chatbot_interface,
+        [chatbot_input, history_output, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type, selector],
+        [history_output, response_output]
+    )
+
+    csv_interface = gr.Interface(
+        lambda url: display_csv(storage_location, url),
+        [csv_input],
+        [csv_output]
+    )
+
+    rss_interface = gr.Interface(
+        generate_rss,
+        [rss_input, storage_location],
+        [rss_output]
+    )
+
+    chatbot_interface.launch()
+    csv_interface.launch()
+    rss_interface.launch()
 
 if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch()
+    main()