Update app.py

app.py

@@ -14,8 +14,7 @@ from huggingface_hub import InferenceClient
 # Check for Hugging Face API key
 api_key = os.getenv('access')
 if not api_key:
-    print("Error: Hugging Face API key not found. Please set the 'access' environment variable.")
-    exit(1)
+    raise EnvironmentError("Error: Hugging Face API key not found. Please set the 'access' environment variable.")
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -96,24 +95,33 @@ def monitor_urls(storage_location, url1, url2, scrape_interval, content_type):
                 driver.quit()
 
 # Define main function to handle user input
-def handle_input(storage_location, url1, url2, scrape_interval, content_type):
+def handle_input(task, storage_location, url1, url2, scrape_interval, content_type):
     global current_task, history
 
-    current_task = f"Monitoring URLs: {url1}, {url2}"
-    history.append(f"Task started: {current_task}")
-    monitor_urls(storage_location, url1, url2, scrape_interval, content_type)
-    return TASK_PROMPT.format(task=current_task, history="\n".join(history))
+    if task == "Monitor URLs for Changes":
+        current_task = f"Monitoring URLs: {url1}, {url2}"
+        history.append(f"Task started: {current_task}")
+        monitor_urls(storage_location, url1, url2, scrape_interval, content_type)
+        return TASK_PROMPT.format(task=current_task, history="\n".join(history))
+    elif task == "Chat with a Friendly Chatbot":
+        return "You can chat with the friendly chatbot below."
 
 # Define the chat response function
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=api_key)
 
 def respond(
     message,
-    history: list[tuple[str, str]],
+    history,
     system_message,
     max_tokens,
     temperature,
     top_p,
+    task,
+    storage_location,
+    url1,
+    url2,
+    scrape_interval,
+    content_type
 ):
     messages = [{"role": "system", "content": system_message}]
 
@@ -139,6 +147,9 @@ def respond(
         response += token
         yield response
 
+    # Handle the selected task
+    handle_input(task, storage_location, url1, url2, scrape_interval, content_type)
+
 # Create Gradio interface
 demo = gr.ChatInterface(
     respond,
@@ -153,6 +164,7 @@ demo = gr.ChatInterface(
             step=0.05,
             label="Top-p (nucleus sampling)",
         ),
+        gr.Dropdown(choices=["Monitor URLs for Changes", "Chat with a Friendly Chatbot"], label="Select Task"),
         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"),