Update app.py

app.py

@@ -1,51 +1,55 @@
+import os
 import gradio as gr
 import requests
-import os
 
-# Function to call CodePal's Code Generator API
-def generate_code(instructions, language, generation_type):
-    api_url = 'https://api.codepal.ai/v1/code-generator/query'
-    api_key = os.getenv('CODEPAL_API_KEY')
-    headers = {'Authorization': f'Bearer {api_key}'}
-    data = {
-        'instructions': instructions,
-        'language': language,
-        'generation_type': generation_type
-    }
+# Retrieve the API key from the environment variable
+api_key = os.getenv("CODEPAL_API_KEY")
+if not api_key:
+    raise ValueError("API key not found. Please set the CODEPAL_API_KEY environment variable.")
+
+# Function to call CodePal's Code Reviewer API
+def review_code(code):
+    api_url = "https://api.codepal.ai/v1/code-reviewer/query"
+    headers = {"Authorization": f"Bearer {api_key}"}
+    data = {"code": code}
     response = requests.post(api_url, headers=headers, data=data)
     if response.status_code == 201:
-        return response.json().get('result', 'No code generated.')
+        return response.json().get("result", "No review available.")
     elif response.status_code == 401:
-        return 'Unauthorized: Invalid or missing API key.'
+        return "Unauthorized: Invalid or missing API key."
     else:
-        return f'Error {response.status_code}: {response.text}'
+        return f"Error {response.status_code}: {response.text}"
 
-# Gradio interface components
+# Build the Gradio interface using Blocks for better layout control
 with gr.Blocks() as demo:
-    gr.Markdown("# Code Generator Interface\nGenerate code snippets using CodePal's Code Generator API.")
+    gr.Markdown("# Code Reviewer Interface\nGet a professional code review for any piece of code.")
+
+    # Hugging Face Login Button (requires `hf_oauth: true` in your README metadata)
+    login_button = gr.LoginButton()
+    
     with gr.Row():
         with gr.Column():
-            login_button = gr.LoginButton()
-            instructions_input = gr.Textbox(label='Function Description', placeholder='Describe the function to generate...')
-            language_input = gr.Dropdown(label='Programming Language', choices=['python', 'javascript', 'go', 'java', 'csharp'])
-            generation_type_input = gr.Radio(label='Generation Type', choices=['minimal', 'standard', 'documented'], value='standard')
-            generate_button = gr.Button(value='Generate Code')
-            clear_button = gr.Button(value='Clear')
-            error_output = gr.Textbox(label='Error Message', visible=False)
+            code_input = gr.Textbox(
+                label="Code to Review", 
+                placeholder="Paste your code here...", 
+                lines=15
+            )
+            review_button = gr.Button("Review Code")
+            clear_button = gr.Button("Clear")
         with gr.Column():
-            output_display = gr.Code(label='Generated Code')
-
-    # Example inputs
-    examples = [
-        ["Function to add two numbers", "python", "standard"],
-        ["Class definition for a linked list", "java", "documented"],
-        ["Script to fetch data from an API", "javascript", "minimal"]
-    ]
-    gr.Examples(examples=examples, inputs=[instructions_input, language_input, generation_type_input])
-
-    # Define button actions
-    generate_button.click(generate_code, inputs=[instructions_input, language_input, generation_type_input], outputs=[output_display, error_output])
-    clear_button.click(lambda: ("", "python", "standard", "", ""), outputs=[instructions_input, language_input, generation_type_input, output_display, error_output])
+            review_output = gr.Code(label="Code Review", language="markdown")
+    
+    # Examples to guide users
+    gr.Examples(
+        examples=[
+            ["def factorial(n):\n    if n == 0:\n        return 1\n    else:\n        return n * factorial(n-1)\n\nprint(factorial(5))"]
+        ],
+        inputs=code_input
+    )
+    
+    # Define the actions for the buttons
+    review_button.click(review_code, inputs=code_input, outputs=review_output)
+    clear_button.click(lambda: ("", ""), outputs=[code_input, review_output])
 
 # Launch the interface
 demo.launch()