Update app.py

app.py

@@ -109,15 +109,21 @@ Start each section with its number and title."""
         response_start += len(code_to_analyze)
         ai_response = full_response[response_start:].strip()
         
-        # Remove any leading text before the first numbered section
-        first_section = ai_response.find("1.")
-        if first_section != -1:
-            ai_response = ai_response[first_section:]
+        # Find the second occurrence of "1. Understanding of the code"
+        first_occurrence = ai_response.find("1. Understanding of the code")
+        if first_occurrence != -1:
+            second_occurrence = ai_response.find("1. Understanding of the code", first_occurrence + 1)
+            if second_occurrence != -1:
+                ai_response = ai_response[second_occurrence:]
+            else:
+                # If we can't find a second occurrence, start from the first one
+                ai_response = ai_response[first_occurrence:]
         
         return ai_response
     else:
         return "Error: Unable to extract the AI's response. Here's the full output:\n\n" + full_response
 
+
 # Create Gradio interface
 iface = gr.Interface(
     fn=code_review,