change code to check to see if answer choices in question

app.py

@@ -29,15 +29,46 @@ def ui_search(query, num_results=3):
         return "Enter a query"
     try:
         r = search(query, num_results)
-        
-        # DEBUG: Show what's in metadata
-        print("METADATA KEYS:", r['metadatas'][0][0].keys())
-        print("FULL METADATA:", r['metadatas'][0][0])
-        
         out = ""
         for i in range(len(r['documents'][0])):
             out += f"\n{'='*60}\nExample {i+1}\n{'='*60}\n"
-            out += r['documents'][0][i] + f"\n\nAnswer: {r['metadatas'][0][i].get('answer', 'N/A')}\n"
+            
+            # Get the full question text
+            question_text = r['documents'][0][i]
+            
+            # Parse question and answer choices
+            import re
+            # Look for answer choices pattern (A. or A) followed by text)
+            lines = question_text.split('\n')
+            question_part = []
+            choices_part = []
+            in_choices = False
+            
+            for line in lines:
+                # Check if line starts with A-E followed by . or )
+                if re.match(r'^[A-E][\.\)]', line.strip()):
+                    in_choices = True
+                    choices_part.append(line)
+                elif in_choices:
+                    # Continue collecting choices if they span multiple lines
+                    if line.strip() and not re.match(r'^[A-E][\.\)]', line.strip()):
+                        choices_part[-1] += " " + line.strip()
+                    elif re.match(r'^[A-E][\.\)]', line.strip()):
+                        choices_part.append(line)
+                else:
+                    question_part.append(line)
+            
+            # Display question
+            out += '\n'.join(question_part).strip() + "\n\n"
+            
+            # Display choices if found
+            if choices_part:
+                out += "Answer Choices:\n"
+                for choice in choices_part:
+                    out += choice.strip() + "\n"
+                out += "\n"
+            
+            out += f"Correct Answer: {r['metadatas'][0][i].get('answer', 'N/A')}\n"
             out += f"Similarity: {1 - r['distances'][0][i]:.3f}\n"
         return out
     except Exception as e: