Update app.py

app.py

@@ -2,17 +2,27 @@ import os
 import gradio as gr
 from groq import Groq
 
-# Initialize Groq client
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# Get API key from Hugging Face Secrets
+api_key = os.environ.get("GROQ_API_KEY")
+
+# Initialize client only if key exists
+if api_key:
+    client = Groq(api_key=api_key)
+else:
+    client = None
+
 
 def review_code(code):
     if not code.strip():
         return "Please paste some code."
 
+    if client is None:
+        return "Groq API key not found. Please add GROQ_API_KEY in Hugging Face Space Secrets."
+
     prompt = f"""
-You are a senior software engineer acting as a code reviewer.
+You are an expert software engineer acting as a code reviewer.
 
-Review the following code and provide feedback in these categories:
+Analyze the following code and give feedback on:
 
 1. Readability
 2. Structure
@@ -25,21 +35,27 @@ Code:
 {code}
 """
 
-    completion = client.chat.completions.create(
-        model="llama3-8b-8192",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.2,
-    )
+    try:
+        completion = client.chat.completions.create(
+            model="llama3-8b-8192",
+            messages=[
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0.2
+        )
+
+        return completion.choices[0].message.content
 
-    return completion.choices[0].message.content
+    except Exception as e:
+        return f"Error while reviewing code: {str(e)}"
 
 
-interface = gr.Interface(
+demo = gr.Interface(
     fn=review_code,
     inputs=gr.Code(label="Paste your code here", language="python"),
     outputs=gr.Markdown(label="AI Code Review"),
     title="Smart Code Reviewer",
-    description="AI assistant that reviews code for readability, structure, maintainability, and best practices.",
+    description="AI assistant that reviews code for readability, structure, maintainability, and best practices."
 )
 
-interface.launch()
+demo.launch()