Update app.py

app.py

@@ -2,54 +2,45 @@ import os
 import gradio as gr
 from groq import Groq
 
-# Initialize Groq Client
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+
+if not GROQ_API_KEY:
+    raise ValueError("Please set GROQ_API_KEY in Hugging Face Secrets")
+
 client = Groq(api_key=GROQ_API_KEY)
 
 
 def review_code(code, language):
+
     if code.strip() == "":
-        return "Please paste some code to review."
+        return "Please paste some code."
 
     prompt = f"""
-You are an expert senior software engineer performing a professional code review.
-
-Analyze the following {language} code based on:
+You are a senior software engineer performing a professional code review.
 
-1. Readability
-2. Structure and modularity
-3. Maintainability
-4. Best practices
-5. Potential bugs
-6. Performance improvements
+Analyze the following {language} code.
 
-Return the response in the following format:
+Return:
 
-Code Quality Scores (0-10):
-Readability:
-Structure:
-Maintainability:
-Best Practices:
+Code Quality Scores (0-10)
+Readability
+Structure
+Maintainability
+Best Practices
 
-Issues Found:
-- bullet points
+Issues Found
+Suggested Improvements
+Refactored Code
 
-Suggested Improvements:
-- bullet points
-
-Refactored Code:
-Provide improved version of the code.
-
-Code to review:
+Code:
 {code}
 """
 
     try:
         completion = client.chat.completions.create(
-            model="llama3-70b-8192",
+            model="llama3-8b-8192",
             messages=[
-                {"role": "system", "content": "You are a helpful code review assistant."},
-                {"role": "user", "content": prompt},
+                {"role": "user", "content": prompt}
             ],
             temperature=0.2,
         )
@@ -57,64 +48,28 @@ Code to review:
         return completion.choices[0].message.content
 
     except Exception as e:
-        return f"Error occurred: {str(e)}"
-
-
-def explain_code(code):
-    if code.strip() == "":
-        return "Please paste some code."
-
-    prompt = f"""
-Explain the following code in a simple and clear way.
-Describe what the code does step-by-step.
-
-Code:
-{code}
-"""
-
-    try:
-        completion = client.chat.completions.create(
-            model="llama3-70b-8192",
-            messages=[
-                {"role": "user", "content": prompt}],
-            temperature=0.3,
-        )
-
-        return completion.choices[0].message.content
-
-    except Exception as e:
-        return f"Error occurred: {str(e)}"
+        return str(e)
 
 
 with gr.Blocks(title="Smart Code Reviewer") as demo:
 
-    gr.Markdown("# 🤖 Smart Code Reviewer")
-    gr.Markdown(
-        "AI-powered assistant that reviews your code for readability, structure, maintainability, and best practices."
-    )
-
-    with gr.Row():
-
-        with gr.Column():
+    gr.Markdown("# Smart Code Reviewer")
+    gr.Markdown("AI assistant that reviews code before human review.")
 
-            language = gr.Dropdown(
-                ["Python", "JavaScript", "Java", "C++", "Go", "Other"],
-                value="Python",
-                label="Programming Language"
-            )
-
-            code_input = gr.Code(
-                label="Paste Your Code",
-                language="python",
-                lines=20
-            )
+    language = gr.Dropdown(
+        ["Python", "JavaScript", "Java", "C++", "Other"],
+        value="Python",
+        label="Programming Language"
+    )
 
-            review_btn = gr.Button("🔍 Review Code", variant="primary")
-            explain_btn = gr.Button("📖 Explain Code")
+    code_input = gr.Textbox(
+        label="Paste Code",
+        lines=20
+    )
 
-        with gr.Column():
+    review_btn = gr.Button("Review Code")
 
-            output = gr.Markdown(label="AI Review Output")
+    output = gr.Markdown()
 
     review_btn.click(
         fn=review_code,
@@ -122,26 +77,6 @@ with gr.Blocks(title="Smart Code Reviewer") as demo:
         outputs=output
     )
 
-    explain_btn.click(
-        fn=explain_code,
-        inputs=code_input,
-        outputs=output
-    )
-
-    gr.Markdown(
-        """
----
-### Features
-- AI-powered code review
-- Code quality scoring
-- Bug detection
-- Refactoring suggestions
-- Code explanation
-
-Built with **Groq + Llama3 + Gradio**
-"""
-    )
-
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0")