Update app.py

app.py

@@ -1,13 +1,14 @@
-# -*- coding: utf-8 -*-
-
 import gradio as gr
 from transformers import pipeline
+import time
 
 # Load the text generation model (GPT-2)
 text_generator = pipeline('text-generation', model='gpt2')
 
 def get_completion(code_snippet):
-    # Code examples to illustrate the output
+    print("Received code snippet, starting processing...")
+    start_time = time.time()
+    
     python_code_examples = """
     ---------------------
     Example 1:
@@ -18,7 +19,7 @@ def get_completion(code_snippet):
     foo()
     print(x)
     Correct output: 5
-    Explanation: The global keyword allows foo() to modify the global variable x.
+    Explanation: The global keyword allows foo() to modify the global variable x, so print(x) outputs 5.
 
     ---------------------
     Example 2:
@@ -29,12 +30,19 @@ def get_completion(code_snippet):
     modify_list(my_list)
     print(my_list)
     Correct output: [0, 4]
-    Explanation: The function appends 4 to input_list, which modifies my_list directly.
+    Explanation: input_list is modified by appending 4, but the reassignment does not affect the original list.
+
+    ---------------------
+    Example 3:
+    print('hello')
+    Correct output: hello
+    Explanation: This line outputs the string 'hello' to the console.
     ---------------------
     """
     
     prompt = f"""
-    Your task is to explain the following code snippet step-by-step, using the examples below as reference.
+    Your task is to explain the following code snippet step-by-step.
+    Use the following examples as references for the explanations:
     {python_code_examples}
     Code Snippet:
     ```
@@ -43,11 +51,15 @@ def get_completion(code_snippet):
     """
 
     try:
-        # Generate text using the Hugging Face model
+        print("Generating text...")
         response = text_generator(prompt, max_length=300, num_return_sequences=1)
+        print("Text generation complete.")
         return response[0]['generated_text']
     except Exception as e:
         return f"An error occurred: {str(e)}"
+    finally:
+        elapsed_time = time.time() - start_time
+        print(f"Processing time: {elapsed_time:.2f} seconds")
 
 # Define the Gradio interface
 iface = gr.Interface(