Update app.py

app.py

@@ -88,21 +88,27 @@ Aber ich kenne mich super aus mit:
 - Anfahrt & Kontakt 📍
 
 Frag mich einfach danach!"""
+
+    # Kategorien prüfen
+    query_text = results[0].page_content.lower()
     
-    best_result = results[0]
-    parts = best_result.page_content.split('\n')
-    category = parts[0].replace('Category: ', '')
-    answer = parts[2].replace('Answer: ', '')
+    # Adress-/Kontaktanfragen
+    if 'kontakt' in query_text or 'wo finde' in query_text or 'adresse' in query_text:
+        for result in results:
+            if 'contact' in result.metadata.get('source', ''):
+                return f"Cool, dass Du zu uns kommen möchtest! Hier findest Du uns:\n\n{result.page_content}\n\nBis bald! 🚀"
+                
+    # Geburtstagsfragen
+    if 'geburtstag' in query_text or 'feiern' in query_text:
+        birthday_info = []
+        for result in results:
+            if 'geburtstagsinfos' in result.metadata.get('source', '').lower():
+                birthday_info.append(result.page_content)
+        if birthday_info:
+            return "Hier sind alle wichtigen Infos zu Kindergeburtstagen:\n\n• " + "\n\n• ".join(birthday_info)
     
-    # Format based on category
-    if 'oeffnungszeiten' in category.lower():
-        return f"Hey! Schau mal, wann Du bei uns vorbeikommen kannst:\n\n{answer}\n\nWir freuen uns auf Dich! 🎉"
-    elif 'hallenregeln' in category.lower():
-        return f"Hey! Damit alle Spaß haben und sicher springen können:\n\n{answer}\n\nDanke, dass Du die Regeln beachtest! 🙌"
-    elif 'kontakt' in category.lower():
-        return f"Cool, dass Du zu uns kommen möchtest!\n\n{answer}\n\nBis bald! 🚀"
-    else:
-        return answer
+    # Standardantwort mit dem besten Match
+    return results[0].page_content
 
 def chat(message, history):
     """Main chat function"""