Update app.py

app.py

@@ -3,16 +3,32 @@ import gradio as gr
 import requests
 
 # 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.")
+def get_api_key():
+    """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.")
+    return api_key
 
 # Function to call CodePal's Code Reviewer API
 def review_code(code):
+    """
+    Sends the provided code to the CodePal Code Reviewer API and returns the review results.
+
+    Parameters:
+    code (str): The code to be reviewed.
+
+    Returns:
+    str: The result of the code review, or an error message if the API request fails.
+    """
     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)
+    headers = {"Authorization": f"Bearer {get_api_key()}"}
+    response = requests.post(api_url, headers=headers, json={"code": code})
+    
+    return handle_api_response(response)
+
+def handle_api_response(response):
+    """Handle the API response and return the appropriate message."""
     if response.status_code == 201:
         return response.json().get("result", "No review available.")
     elif response.status_code == 401:
@@ -21,35 +37,33 @@ def review_code(code):
         return f"Error {response.status_code}: {response.text}"
 
 # Build the Gradio interface using Blocks for better layout control
-with gr.Blocks() as demo:
-    gr.Markdown("# Code Reviewer Interface\nGet a professional code review for any piece of code.")
+def create_interface():
+    """Create the Gradio interface for the code reviewer."""
+    with gr.Blocks() as demo:
+        gr.Markdown("# Code Reviewer Interface\nGet a professional code review for any piece of code.")
+        gr.LoginButton()
 
-    # Hugging Face Login Button (requires `hf_oauth: true` in your README metadata)
-    login_button = gr.LoginButton()
-    
-    with gr.Row():
-        with gr.Column():
-            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():
-            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])
+        with gr.Row():
+            with gr.Column():
+                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():
+                review_output = gr.Code(label="Code Review", language="markdown")
+
+        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
+        )
+
+        review_button.click(review_code, inputs=code_input, outputs=review_output)
+        clear_button.click(lambda: ("", ""), outputs=[code_input, review_output])
+
+    return demo
 
-# Launch the interface
-demo.launch()
+# Launch the Gradio interface
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch()