Update agent.py

agent.py

@@ -53,7 +53,7 @@ def monitor_urls(storage_location, urls, scrape_interval, content_type):
                         else:
                             current_content = driver.page_source
                         current_hash = hashlib.md5(str(current_content).encode('utf-8')).hexdigest()
-                        if current_hash != previous_hashes[url]:
+                        if current_hash!= previous_hashes[url]:
                             previous_hashes[url] = current_hash
                             date_time_str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
                             HISTORY.append(f"Change detected at {url} on {date_time_str}")
@@ -150,4 +150,44 @@ def generate_rss_feed(url):
 def chat_interface(message, history, system_message, max_tokens, temperature, top_p, storage_location, urls, scrape_interval, content_type):
     response = respond(message, history, system_message, max_tokens, temperature, top_p)
     history.append((message, response))
-    return history, response
+    return history, response
+
+if __name__ == "__main__":
+    # Define the Gradio interface
+    import gradio as gr
+
+    def create_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")
+                    start_button = gr.Button("Start Scraping")
+                    csv_output = gr.Textbox(label="CSV Output", interactive=False)
+
+                with gr.Column():
+                    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])
+
+            # 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)
+
+            rss_button.click(generate_rss_feed, inputs=[selected_url], outputs=rss_output)
+
+        return demo
+
+    demo = create_interface()
+    demo.launch()