Update app.py

app.py

@@ -1,4 +1,5 @@
 import groq
+from groq import Groq, AuthenticationError, RateLimitError
 import gradio as gr
 import re
 import json
@@ -6,27 +7,30 @@ import os
 
 
 # Setup Groq
-client = groq.Client(api_key=os.environ.get("GROQ_API_KEY"))  # Correct initialization
+client = groq.Client(api_key=os.environ.get("GROQ_API_KEY")) 
 model = "llama-3.3-70b-versatile"
 
 def handle_groq_error(e, model_name):
     error_data = e.args[0]
-    
+
     if isinstance(error_data, str):
-        # Use regex to extract the JSON part of the string
         json_match = re.search(r'(\{.*\})', error_data)
         if json_match:
             json_str = json_match.group(1)
-            # Ensure the JSON string is well-formed
-            json_str = json_str.replace("'", '"')  # Replace single quotes with double quotes
+            json_str = json_str.replace("'", '"') 
             error_data = json.loads(json_str)
 
-    if isinstance(e, groq.RateLimitError):
+    if isinstance(e, AuthenticationError):
+        if isinstance(error_data, dict) and 'error' in error_data and 'message' in error_data['error']:
+            error_message = error_data['error']['message']
+            raise discord.app_commands.AppCommandError(error_message)
+    elif isinstance(e, RateLimitError):
         if isinstance(error_data, dict) and 'error' in error_data and 'message' in error_data['error']:
             error_message = error_data['error']['message']
-            raise gr.Error(error_message)
+            error_message = re.sub(r'org_[a-zA-Z0-9]+', 'org_(censored)', error_message) 
+            raise discord.app_commands.AppCommandError(error_message)
     else:
-        raise gr.Error(f"Error during Groq API call: {e}")
+        raise discord.app_commands.AppCommandError(f"Error during Groq API call: {e}")
 
 
 
@@ -42,12 +46,17 @@ def calculate_and_roast(expression):
         # Groq API call
         try:
             prompt = f"Roast me for calculating something as simple as '{expression}'. Make it funny."
-            response = client.request(model, {"prompt": prompt})
-            roast = response["text"][0]
+            chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model=model
+            )
+            roast = chat_completion.choices[0].message.content
             return result, roast
 
-        except groq.GroqApiException as e:
-             handle_groq_error(e, model)
+        except AuthenticationError as e:
+            handle_groq_error(e, model)
+        except RateLimitError as e:
+            handle_groq_error(e, model)
 
     except (ValueError, TypeError, SyntaxError, ZeroDivisionError) as e:
         return f"Calculation error: {e}", None