Update app.py

app.py

@@ -12,14 +12,25 @@ import re
 import base64
 import logging
 import os
+import sys
 
 # Import OpenAI library
 import openai
 
-# Set up logging
-logging.basicConfig(filename='app.log', level=logging.INFO,
-                    format='%(asctime)s %(levelname)s %(name)s %(message)s')
+# Set up logging to output to the console
 logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+
+# Create a console handler
+console_handler = logging.StreamHandler(sys.stdout)
+console_handler.setLevel(logging.INFO)
+
+# Create a formatter and set it for the handler
+formatter = logging.Formatter('%(asctime)s %(levelname)s %(name)s %(message)s')
+console_handler.setFormatter(formatter)
+
+# Add the handler to the logger
+logger.addHandler(console_handler)
 
 # Initialize models and variables
 logger.info("Initializing models and variables")
@@ -56,13 +67,13 @@ CATEGORIES = [
 
 # Set up Groq Cloud API key and base URL
 GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+
 if not GROQ_API_KEY:
     logger.error("GROQ_API_KEY environment variable not set.")
-    raise ValueError("Please set the GROQ_API_KEY environment variable.")
 
 # Set OpenAI API key and base URL to use Groq Cloud API
 openai.api_key = GROQ_API_KEY
-openai.api_base = "https://api.groq.com/openai/v1"
+openai.api_base = "https://api.groq.com/openai/v1/chat/completions"  # Updated endpoint
 
 # Function to parse bookmarks from HTML
 def parse_bookmarks(file_content):
@@ -307,6 +318,10 @@ def process_uploaded_file(file):
 
 # Chatbot response using Groq Cloud API
 def chatbot_response(user_query):
+    if not GROQ_API_KEY:
+        logger.warning("GROQ_API_KEY not set.")
+        return "API key not set. Please set the GROQ_API_KEY environment variable in the Hugging Face Space settings."
+
     if not bookmarks:
         logger.warning("No bookmarks available for chatbot")
         return "No bookmarks available. Please upload and process your bookmarks first."
@@ -336,7 +351,7 @@ Please identify the most relevant bookmarks that match the user's query. Provide
 
         # Call the Groq Cloud API via the OpenAI client
         response = openai.ChatCompletion.create(
-            model='llama3-8b-8192',  # Specify the Groq Cloud model
+            model='llama3-8b-8192',  # Verify this model name with Groq Cloud API documentation
             messages=[
                 {"role": "system", "content": "You help users find relevant bookmarks based on their queries."},
                 {"role": "user", "content": prompt}
@@ -351,8 +366,10 @@ Please identify the most relevant bookmarks that match the user's query. Provide
         return answer
 
     except Exception as e:
-        logger.error(f"Error in chatbot response generation: {e}")
-        return "Error processing your query."
+        error_message = f"Error processing your query: {str(e)}"
+        logger.error(error_message)
+        print(error_message)  # Ensure error appears in Hugging Face Spaces logs
+        return error_message
 
 # Edit a bookmark
 def edit_bookmark(bookmark_idx, new_title, new_url, new_category):