Update gemini_agent.py

gemini_agent.py

@@ -265,10 +265,17 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         
         # If the question expects a single value, extract it
         if any(kw in question.lower() for kw in ["how many", "what is the", "who", "where", "give only", "provide only"]):
-            # Extract the first number, word, or phrase (tweak regex as needed)
-            match = re.search(r'^[A-Za-z0-9 ,+-]+', answer)
-            if match:
-                answer = match.group(0).strip()
+            # Special handling for "give only the first name"
+            if "give only the first name" in question.lower() or "first name" in question.lower():
+                # Extract first name from full name
+                name_match = re.search(r'\b([A-ZĄĆĘŁŃÓŚŹŻ][a-ząćęłńóśźż]+)', answer)
+                if name_match:
+                    answer = name_match.group(1)
+            else:
+                # Extract the first number, word, or phrase (tweak regex as needed)
+                match = re.search(r'^[A-Za-z0-9 ,+-]+', answer)
+                if match:
+                    answer = match.group(0).strip()
         
         # Post-processing for chess move extraction
         if 'chess position' in question.lower() and 'image' in question.lower():