Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import gradio as gr
 import requests
 
-# Retrieve the API key from the environment variable
 def get_api_key():
     """Retrieve the API key from the environment variable."""
     api_key = os.getenv("CODEPAL_API_KEY")
@@ -10,33 +9,38 @@ def get_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 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:
+        return "Unauthorized: Invalid or missing API key."
+    else:
+        return f"Error {response.status_code}: {response.text}"
+
 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.
+        code (str): The code to be reviewed.
 
     Returns:
-    str: The result of the code review, or an error message if the API request fails.
+        str: The result of the code review, or an error message if the API request fails.
     """
+    if not code.strip():
+        return "Error: The code input is empty. Please provide code for review."
+
     api_url = "https://api.codepal.ai/v1/code-reviewer/query"
     headers = {"Authorization": f"Bearer {get_api_key()}"}
-    response = requests.post(api_url, headers=headers, json={"code": code})
-    
-    return handle_api_response(response)
+    payload = {"code": code}
 
-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:
-        return "Unauthorized: Invalid or missing API key."
-    else:
-        return f"Error {response.status_code}: {response.text}"
+    try:
+        response = requests.post(api_url, headers=headers, json=payload)
+        return handle_api_response(response)
+    except requests.RequestException as e:
+        return f"An error occurred while making the API request: {e}"
 
-# Build the Gradio interface using Blocks for better layout control
 def create_interface():
     """Create the Gradio interface for the code reviewer."""
     with gr.Blocks() as demo:
@@ -63,7 +67,6 @@ def create_interface():
 
     return demo
 
-# Launch the Gradio interface
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()