Update app.py

app.py

@@ -82,33 +82,30 @@ Q18: Is private sector experience allowed?
 Q19: Do references need to be identified?  
 Q20: Is subcontracting permitted?
 
-Answer clearly and in the same format:
-Q1: ...
-A1: ...
-Q2: ...
-A2: ...
 ...
 """
 
 # 🧼 Cleaner
 def clean_output(raw_output):
-    lines = raw_output.splitlines()
-    cleaned_lines = []
-    started = False
-
-    for line in lines:
-        if line.strip().startswith("Q1:"):
-            started = True
-        if started:
-            cleaned_lines.append(line)
-
-    stop_idx = len(cleaned_lines)
-    for i, line in enumerate(cleaned_lines[5:], 5):
-        if "CONTENT:" in line or "You are an expert" in line:
-            stop_idx = i
-            break
-
-    return "\n".join(cleaned_lines[:stop_idx]).strip()
+    # Find first valid Q1
+    start_idx = raw_output.find("Q1:")
+    if start_idx == -1:
+        return raw_output.strip()
+
+    cleaned = raw_output[start_idx:]
+
+    # Remove everything after second instance of Q1 (to drop repeated prompts)
+    second_q1 = cleaned.find("Q1:", 3)  # skip first one
+    if second_q1 != -1:
+        cleaned = cleaned[:second_q1]
+
+    # Drop leftover instructions if they show up later
+    cut_phrases = ["You are an expert", "Now provide answers", "CONTENT:", "Answer clearly and in the same format:"]
+    for phrase in cut_phrases:
+        if phrase in cleaned:
+            cleaned = cleaned.split(phrase)[0]
+
+    return cleaned.strip()
 
 # 🚀 Main analysis function
 @spaces.GPU(duration=150)