Update app.py

app.py

@@ -3,32 +3,36 @@
 import gradio as gr
 import google.generativeai as genai
 
-# Configure API key
+# Configure the API key
 genai.configure(api_key='AIzaSyArybMiPDZARDCz7yZBjzEMx6zXgOXHtoc')
 
-# Define the completion function using the new API methods
+# Define the completion function with error handling
 def get_completion(code_snippet):
-    python_code_examples = """
-    ---------------------
-    Example 1: x = 10 ... (your code examples)
-    ---------------------
-    """
-    prompt = f"""
-    Your task is to explain the following code snippet step-by-step:
-    {python_code_examples}
-    ```
-    {code_snippet}
-    ```
-    """
+    try:
+        python_code_examples = """
+        ---------------------
+        Example 1: x = 10 ... (Your code examples here)
+        ---------------------
+        """
+        prompt = f"""
+        Your task is to explain the following code snippet step-by-step:
+        {python_code_examples}
+        ```
+        {code_snippet}
+        ```
+        """
 
-    # Generate text using the new method
-    response = genai.generate_content(
-        prompt=prompt,
-        temperature=0,
-        max_output_tokens=500,
-    )
+        # Generate content using the new API method
+        response = genai.generate_content(
+            prompt=prompt,
+            temperature=0,
+            max_output_tokens=500,
+        )
+        return response.text
 
-    return response.text
+    except Exception as e:
+        # Return the error message
+        return f"An error occurred: {str(e)}"
 
 # Create the Gradio interface
 iface = gr.Interface(