Update app.py

app.py

@@ -101,7 +101,16 @@ Provide a detailed review including:
 
 Start each section with its number and title."""
 
-    return get_completion(few_shot_prompt, model, tokenizer)
+    full_response = get_completion(few_shot_prompt, model, tokenizer)
+    
+    # Extract only the part of the response after the input code
+    start_marker = f"Now, review the following code using the same approach:\n\n{code_to_analyze}\n\n"
+    start_index = full_response.find(start_marker)
+    if start_index != -1:
+        relevant_response = full_response[start_index + len(start_marker):].strip()
+        return relevant_response
+    else:
+        return "Error: Unable to extract the relevant part of the AI's response."
 
 # Create Gradio interface
 iface = gr.Interface(