Update app.py

app.py

@@ -182,35 +182,50 @@ def respond(
     )
     return response
 
+# Function to start scraping
+def start_scraping(storage_location, url1, url2, scrape_interval, content_type):
+    handle_input(storage_location, url1, url2, scrape_interval, content_type)
+    return f"Started scraping {url1} and {url2} every {scrape_interval} minutes."
+
+# Function to display CSV content
+def display_csv(storage_location):
+    if os.path.exists(storage_location):
+        with open(storage_location, "r") as file:
+            return file.read()
+    else:
+        return "No data available."
+
 # Create Gradio interface
 def chat_interface(message, system_message, max_tokens, temperature, top_p, storage_location, url1, url2, scrape_interval, content_type):
     global history
     response = respond(message, history, system_message, max_tokens, temperature, top_p)
     history.append((message, response))
-    handle_input(storage_location, url1, url2, scrape_interval, content_type)
     return history, ""
 
-demo = gr.Interface(
-    fn=chat_interface,
-    inputs=[
-        gr.Textbox(label="Message"),
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
-        gr.Textbox(value=default_file_path, label="Storage Location"),
-        gr.Textbox(value="https://www.culver.k12.in.us/", label="URL 1"),
-        gr.Textbox(value="https://www.facebook.com/CulverCommunitySchools", label="URL 2"),
-        gr.Slider(minimum=1, maximum=60, value=5, step=1, label="Scrape Interval (minutes)"),
-        gr.Radio(choices=["text", "media", "both"], value="text", label="Content Type"),
-    ],
-    outputs=[
-        gr.Chatbot(label="Chat History"),
-        gr.Textbox(label="Response")
-    ],
-    title="Culvers Site Monitor and Chatbot",
-    description="Monitor changes on Culvers' websites and log them into a CSV file. Also, chat with a friendly chatbot."
-)
+demo = gr.Blocks()
+
+with demo:
+    with gr.Row():
+        with gr.Column():
+            message = gr.Textbox(label="Message")
+            system_message = gr.Textbox(value="You are a friendly Chatbot.", 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=default_file_path, label="Storage Location")
+            url1 = gr.Textbox(value="https://www.culver.k12.in.us/", label="URL 1")
+            url2 = gr.Textbox(value="https://www.facebook.com/CulverCommunitySchools", label="URL 2")
+            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, url1, url2, scrape_interval, content_type], outputs=csv_output)
+    message.submit(chat_interface, inputs=[message, system_message, max_tokens, temperature, top_p, storage_location, url1, url2, scrape_interval, content_type], outputs=[chat_history, response_box])
 
 if __name__ == "__main__":
     demo.launch()