Update app.py

app.py

@@ -2,81 +2,44 @@ import os
 import gradio as gr
 from groq import Groq
 
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+# Initialize Groq client
+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() == "":
+def review_code(code):
+    if not code.strip():
         return "Please paste some code."
 
     prompt = f"""
-You are a senior software engineer performing a professional code review.
+You are a senior software engineer acting as a code reviewer.
 
-Analyze the following {language} code.
+Review the following code and provide feedback in these categories:
 
-Return:
+1. Readability
+2. Structure
+3. Maintainability
+4. Best Practices
 
-Code Quality Scores (0-10)
-Readability
-Structure
-Maintainability
-Best Practices
-
-Issues Found
-Suggested Improvements
-Refactored Code
+Also suggest an improved version of the code.
 
 Code:
 {code}
 """
 
-    try:
-        completion = client.chat.completions.create(
-            model="llama3-8b-8192",
-            messages=[
-                {"role": "user", "content": prompt}
-            ],
-            temperature=0.2,
-        )
-
-        return completion.choices[0].message.content
-
-    except Exception as e:
-        return str(e)
-
-
-with gr.Blocks(title="Smart Code Reviewer") as demo:
-
-    gr.Markdown("# Smart Code Reviewer")
-    gr.Markdown("AI assistant that reviews code before human review.")
-
-    language = gr.Dropdown(
-        ["Python", "JavaScript", "Java", "C++", "Other"],
-        value="Python",
-        label="Programming Language"
+    completion = client.chat.completions.create(
+        model="llama3-8b-8192",
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.2,
     )
 
-    code_input = gr.Textbox(
-        label="Paste Code",
-        lines=20
-    )
+    return completion.choices[0].message.content
 
-    review_btn = gr.Button("Review Code")
-
-    output = gr.Markdown()
-
-    review_btn.click(
-        fn=review_code,
-        inputs=[code_input, language],
-        outputs=output
-    )
 
+interface = 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.",
+)
 
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0")
+interface.launch()