Update app.py

app.py

@@ -23,8 +23,10 @@ def create_interface():
                 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")
@@ -34,21 +36,21 @@ def create_interface():
             """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"])
+                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], outputs=csv_output)
+        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, scrape_interval, content_type)
+                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], outputs=[chat_history, response_box])
+        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():
@@ -59,7 +61,7 @@ def create_interface():
         def generate_rss_feed_callback(values):
             """Generate RSS feed callback"""
             try:
-                generate_rss_feed(values["selected_url"])
+                rss_output.value = generate_rss_feed(storage_location, values["selected_url"])
             except Exception as e:
                 print(f"Error: {str(e)}")