Update app.py

app.py

@@ -1,25 +1,35 @@
 import gradio as gr
 import openai
 import os
+import time
 
 # Ensure the API key is loaded correctly
-openai.api_key = os.getenv("GROQ_API_KEY")
+api_key = os.getenv("GROQ_API_KEY")
+if not api_key:
+    raise ValueError("API key is missing. Please set the GROQ_API_KEY environment variable.")
+openai.api_key = api_key
 openai.api_base = "https://api.groq.com/openai/v1"
 
 def get_groq_response(message):
     try:
-        # Logging the message before sending
+        # Log the message before sending
         print(f"Sending message: {message}")
         
+        # Call the API with a timeout to avoid hanging requests
         response = openai.ChatCompletion.create(
             model="llama-3.3-70b-versatile",
             messages=[
                 {"role": "system", "content": "Precise answer"},
                 {"role": "user", "content": message}
-            ]
+            ],
+            timeout=30  # Set a reasonable timeout (30 seconds)
         )
-        
-        # Correcting the response structure
+
+        # Check if the response structure is as expected
+        if 'choices' not in response or len(response['choices']) == 0:
+            raise ValueError("Unexpected API response structure.")
+
+        # Extract the bot's message
         bot_message = response['choices'][0]['message']['content']
         
         # Log the bot's response
@@ -27,17 +37,41 @@ def get_groq_response(message):
         
         return bot_message
     
+    except openai.error.AuthenticationError as e:
+        # Handle authentication error (invalid API key)
+        print(f"Authentication error: {str(e)}")
+        return "Authentication failed. Please check your API key."
+    
+    except openai.error.RateLimitError as e:
+        # Handle rate limit error (too many requests)
+        print(f"Rate limit exceeded: {str(e)}")
+        return "Rate limit exceeded. Please try again later."
+    
+    except openai.error.APIError as e:
+        # Handle other API-related errors
+        print(f"API error: {str(e)}")
+        return "API error occurred. Please try again later."
+    
+    except openai.error.Timeout as e:
+        # Handle timeout errors
+        print(f"Request timed out: {str(e)}")
+        return "The request timed out. Please try again later."
+    
     except Exception as e:
+        # Handle all other generic errors
         print(f"Error: {str(e)}")
-        return f"Error: {str(e)}"
-
+        return f"An error occurred: {str(e)}"
 
 def chatbot(user_input, history=[]):
+    # Handle empty user input
+    if not user_input.strip():
+        return history, history
+    
+    # Get the bot's response
     bot_response = get_groq_response(user_input)
     history.append((user_input, bot_response))
     return history, history
 
-
 chat_interface = gr.Interface(
     fn=chatbot,
     inputs=["text", "state"],