Update app.py

app.py

@@ -1,84 +1,76 @@
 # -*- coding: utf-8 -*-
 
+# Import necessary libraries
+import gradio as gr
+import google.generativeai as genai
 
+# Configure the API key
+genai.configure(api_key='AIzaSyYourGenAIKeyHere')
 
+# Load the generative model using the GenerativeModel class
+model = genai.GenerativeModel(name="gemini", version="v1")
 
-
-#@title DECIPHER THE CODE EXPLAINER 
-import gradio as gr
-import google.generativeai as palm
-
-# load model
-# PaLM API Key here
-palm.configure(api_key='AIzaSyArybMiPDZARDCz7yZBjzEMx6zXgOXHtoc')
-# Use the palm.list_models function to find available models
-# PaLM 2 available in 4 sizes: Gecko, Otter, Bison and Unicorn (largest)
-models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods]
-model = models[0].name
-# define completion function
+# Define the completion function using generate_content
 def get_completion(code_snippet):
+    python_code_examples = """
+    ---------------------
+    Example 1: Code Snippet
+    x = 10
+    def foo():
+        global x
+        x = 5
+    foo()
+    print(x)
+    Correct output: 5
+    Code Explanation: Inside the foo function, the global keyword is used to modify the global variable x to be 5.
+    So, print(x) outside the function prints the modified value, which is 5.
+    ---------------------
+    Example 2: Code Snippet
+    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: Inside the modify_list function, an element 4 is appended to input_list.
+    Then, input_list is reassigned to a new list [1, 2, 3], but this change doesn't affect the original list.
+    So, print(my_list) outputs [0, 4].
+    ---------------------
+    """
 
-  python_code_examples = f"""
-  ---------------------
-  Example 1: Code Snippet
-  x = 10
-  def foo():
-      global x
-      x = 5
-  foo()
-  print(x)
-  Correct output: 5
-  Code Explanation: Inside the foo function, the global keyword is used to modify the global variable x to be 5.
-  So, print(x) outside the function prints the modified value, which is 5.
-  ---------------------
-  Example 2: Code Snippet
-  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: Inside the modify_list function, an element 4 is appended to input_list.
-  Then, input_list is reassigned to a new list [1, 2, 3], but this change doesn't affect the original list.
-  So, print(my_list) outputs [0, 4].
-  ---------------------
-  """
+    prompt = f"""
+    Your task is to act as any language Code Explainer.
+    I'll give you a Code Snippet.
+    Your job is to explain the Code Snippet step-by-step.
+    Break down the code into as many steps as possible.
+    Share intermediate checkpoints & steps along with results.
+    Few good examples of Python code output between #### separator:
+    ####
+    {python_code_examples}
+    ####
+    Code Snippet is shared below, delimited with triple backticks:
+    ```
+    {code_snippet}
+    ```
+    """
 
-  prompt = f"""
-  Your task is to act as any language Code Explainer.
-  I'll give you a Code Snippet.
-  Your job is to explain the Code Snippet step-by-step.
-  Break down the code into as many steps as possible.
-  Share intermediate checkpoints & steps along with results.
-  Few good examples of Python code output between #### separator:
-  ####
-  {python_code_examples}
-  ####
-  Code Snippet is shared below, delimited with triple backticks:
-  ```
-  {code_snippet}
-  ```
-  """
+    # Generate content using the unified method
+    completion = model.generate_content(
+        prompt=prompt,
+        temperature=0,
+        max_output_tokens=500,
+    )
 
-  completion = palm.generate_text(
-      model=model,
-      prompt=prompt,
-      temperature=0,
-      # The maximum length of the response
-      max_output_tokens=500,
-      )
-  response = completion.result
-  return response
+    response = completion.result
+    return response
 
-# define app UI
-iface = gr.Interface(fn=get_completion, inputs=[gr.Textbox(label="Insert Code Snippet",lines=5)],
-                    outputs=[gr.Textbox(label="Explanation Here",lines=8)],
-                    title="Code Explainer"
-                    )
+# Define the Gradio app UI
+iface = gr.Interface(
+    fn=get_completion,
+    inputs=[gr.Textbox(label="Insert Code Snippet", lines=5)],
+    outputs=[gr.Textbox(label="Explanation Here", lines=8)],
+    title="Code Explainer"
+)
 
 iface.launch()
-
-
-
-