Update app.py

app.py

@@ -1,62 +1,63 @@
 # -*- coding: utf-8 -*-
 
 import gradio as gr
-import google.generativeai as genai
+import openai
+import os
 
-# Configure the API key
-genai.configure(api_key='AIzaSyArybMiPDZARDCz7yZBjzEMx6zXgOXHtoc')
+# Configure the OpenAI API key
+openai.api_key = os.getenv('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')  # Ensure your API key is set in environment variables
 
-# Define the completion function with error handling
 def get_completion(code_snippet):
-    try:
-        python_code_examples = """
-        ---------------------
-        Example 1: x = 10
-        def foo():
-            global x
-            x = 5
-        foo()
-        print(x)
-        Correct output: 5
-        Code Explanation: The global keyword modifies the global variable x to be 5.
-        ---------------------
-        Example 2: def modify_list(input_list):
-            input_list.append(4)
-            input_list = [1, 2, 3]
-        my_list = [0]
-        modify_list(my_list)
-        print(my_list)
-        Correct output: [0, 4]
-        Code Explanation: Appending 4 modifies the input_list but does not affect the original my_list.
-        ---------------------
-        """
-        
-        prompt = f"""
-        Your task is to explain the following code snippet step-by-step:
-        {python_code_examples}
-        ```
-        {code_snippet}
-        ```
-        """
-
-        # Use the correct API method to generate text
-        response = genai.generate_text(
-            prompt=prompt,
-            temperature=0,
-            max_output_tokens=500,
-        )
+    # Code examples to illustrate the output
+    python_code_examples = """
+    ---------------------
+    Example 1:
+    x = 10
+    def foo():
+        global x
+        x = 5
+    foo()
+    print(x)
+    Correct output: 5
+    Explanation: The global keyword allows foo() to modify the global variable x.
 
-        return response.result  # Access the result of the response
+    ---------------------
+    Example 2:
+    def modify_list(input_list):
+        input_list.append(4)
+        input_list = [1, 2, 3]
+    my_list = [0]
+    modify_list(my_list)
+    print(my_list)
+    Correct output: [0, 4]
+    Explanation: The function appends 4 to input_list, which modifies my_list directly.
+    ---------------------
+    """
+    
+    prompt = f"""
+    Your task is to explain the following code snippet step-by-step, using the examples below as reference.
+    {python_code_examples}
+    Code Snippet:
+    ```
+    {code_snippet}
+    ```
+    """
 
+    try:
+        completion = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Specify the model
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=500
+        )
+        return completion.choices[0].message['content']
     except Exception as e:
-        # Return the error message
         return f"An error occurred: {str(e)}"
 
-# Create the Gradio interface
+# Define the Gradio interface
 iface = gr.Interface(
     fn=get_completion,
-    inputs=gr.Textbox(label="Insert Code Snippet", lines=5),
-    outputs=gr.Textbox(label="Explanation Here", lines=8),
+    inputs=[gr.Textbox(label="Insert Code Snippet", lines=5)],
+    outputs=[gr.Textbox(label="Explanation Here", lines=8)],
     title="Code Explainer"
 )