Upload main.py

main.py

@@ -7,10 +7,9 @@ from datetime import datetime
 
 app = FastAPI()
 
-# --- KONFIGURATION ---
+# --- SETUP ---
 COLLECTION_KNOWLEDGE = "knowledge_base"
 COLLECTION_RULES = "availability_rules"
-COLLECTION_INBOX = "inbox"
 KNOWLEDGE_CACHE = []
 
 # --- FIREBASE VERBINDUNG ---
@@ -31,50 +30,31 @@ except Exception as e:
 # --- CACHE LADEN ---
 def reload_knowledge():
     global KNOWLEDGE_CACHE
-    if not db: return 0
+    if not db: return
     try:
         docs = db.collection(COLLECTION_KNOWLEDGE).stream()
         KNOWLEDGE_CACHE = [d.to_dict() for d in docs]
         print(f"📚 {len(KNOWLEDGE_CACHE)} Einträge geladen.")
-        return len(KNOWLEDGE_CACHE)
     except Exception as e:
         print(f"❌ Cache Fehler: {e}")
-        return 0
 
 @app.on_event("startup")
 async def startup():
     reload_knowledge()
 
 def get_stem(word):
+    # Einfaches Stemming für bessere Treffer
     w = word.lower().strip()
     for end in ["ern", "en", "er", "es", "st", "te", "e", "s", "t"]:
         if w.endswith(end) and len(w) > len(end)+2: return w[:-len(end)]
     return w
 
 # ==========================================
-# 1. HAUPTSEITE & MONITORING (Repariert!)
-# ==========================================
-@app.get("/")
-def home():
-    # Hier ist die Info zurück!
-    return {
-        "status": "Online", 
-        "docs_loaded": len(KNOWLEDGE_CACHE),
-        "endpoints": ["/check_availability", "/search", "/refresh_knowledge"]
-    }
-
-@app.get("/refresh_knowledge")
-def refresh():
-    count = reload_knowledge()
-    return {"status": "Cache aktualisiert", "docs_loaded": count}
-
-# ==========================================
-# 2. TOOL: VERFÜGBARKEIT PRÜFEN
+# TOOL 1: VERFÜGBARKEIT (Bleibt wie es ist)
 # ==========================================
 @app.post("/check_availability")
 async def check_availability(request: Request):
     print("🚦 TOOL CALL: checkAvailability")
-    
     today = datetime.now().strftime("%Y-%m-%d")
     status = "available"
     message = "Normaler Betrieb."
@@ -85,16 +65,13 @@ async def check_availability(request: Request):
             for r in rules:
                 rd = r.to_dict()
                 if rd.get('start_date') <= today <= rd.get('end_date'):
-                    print(f"🛑 REGEL GEFUNDEN: {rd.get('name')}")
+                    print(f"🛑 REGEL AKTIV: {rd.get('name')}")
                     status = "unavailable"
                     message = rd.get('instruction_text')
                     break
     except Exception as e:
         print(f"❌ ERROR CHECK: {e}")
-        return {"result": json.dumps({"status": "available", "instruction": "Fehler ignoriert, normal arbeiten."})}
 
-    print(f"👉 RESULTAT: {status} | {message}")
-    
     return {
         "result": json.dumps({
             "status": status,
@@ -103,7 +80,7 @@ async def check_availability(request: Request):
     }
 
 # ==========================================
-# 3. TOOL: WISSENSSUCHE
+# TOOL 2: SUCHE (Jetzt mit Udo & Capaneo Priorität)
 # ==========================================
 @app.post("/search")
 async def search(request: Request):
@@ -111,6 +88,7 @@ async def search(request: Request):
         data = await request.json()
         query = ""
         
+        # Parsing
         if "search_query" in data: query = data["search_query"]
         elif "message" in data and "toolCalls" in data["message"]:
             args = data["message"]["toolCalls"][0]["function"]["arguments"]
@@ -120,44 +98,69 @@ async def search(request: Request):
         print(f"🔎 FRAGE: '{query}'")
         if not query: return {"result": "Akustik-Fehler."}
 
-        STOP_WORDS = ["capaneo", "udo", "schober", "firma", "gmbh", "hallo", "demo"]
+        # --- UPDATE: STOP WÖRTER BEREINIGT ---
+        # Wir entfernen "capaneo" und "udo" hier! Sie sind wichtig!
+        # Nur Füllwörter bleiben gesperrt.
+        STOP_WORDS = ["hallo", "guten", "tag", "moin", "bitte", "danke", "frage", "kannst", "sagen"]
+        
         q_words = [get_stem(w) for w in query.lower().split() if len(w)>2]
         relevant_words = [w for w in q_words if w not in STOP_WORDS]
-        
-        if not relevant_words:
-            return {"result": "Dazu habe ich keine Informationen."}
 
+        if not relevant_words:
+             return {"result": "Dazu habe ich keine Informationen."}
+        
         best_doc = None
         best_score = 0
 
         for doc in KNOWLEDGE_CACHE:
             score = 0
-            txt = (doc.get("question", "") + " " + doc.get("answer", "")).lower()
-            kws = [k.lower() for k in doc.get("keywords", [])]
+            # Wir holen Titel (Frage) und Inhalt (Antwort)
+            title = doc.get("question", "").lower()
+            content = doc.get("answer", "").lower()
+            keywords = [k.lower() for k in doc.get("keywords", [])]
             
-            for k in kws:
-                k_stem = get_stem(k)
-                if k_stem in relevant_words: score += 30
-            for w in relevant_words:
-                if w in txt: score += 5
+            for word in relevant_words:
+                word_stem = get_stem(word)
+
+                # 1. TITEL TREFFER (Das Wichtigste!)
+                # Wenn der Kunde "Capaneo" sagt und im Titel steht "Wer ist Capaneo" -> BINGO!
+                if word in title: 
+                    score += 50 
+                
+                # 2. KEYWORD TREFFER
+                # Keywords sind auch stark
+                for k in keywords:
+                    if get_stem(k) == word_stem:
+                        score += 30
+                
+                # 3. TEXT TREFFER (Nur Beiwerk)
+                # Verhindert, dass jedes Dokument mit "Capaneo" im Footer gewinnt
+                if word in content:
+                    score += 5
 
             if score > best_score:
                 best_score = score
                 best_doc = doc
 
-        if best_doc and best_score >= 15:
-            print(f"🏆 TREFFER ({best_score}): {best_doc.get('question')}")
+        # Score Schwelle auf 20 (damit "Hallo" keine Treffer bringt, aber "Udo" schon)
+        if best_doc and best_score >= 20:
+            print(f"🏆 TREFFER ({best_score} Pkt): {best_doc.get('question')}")
             return {"result": best_doc.get("answer")}
         else:
-            return {"result": "Dazu habe ich keine Informationen."}
+            print(f"⚠️ Zu wenig Relevanz (Max Score: {best_score})")
+            return {"result": "Dazu habe ich leider keine spezifischen Informationen in meiner Datenbank."}
 
     except Exception as e:
         print(f"❌ SEARCH ERROR: {e}")
-        return {"result": "Systemfehler."}
+        return {"result": "Fehler."}
 
 # ==========================================
-# 4. MÜLLSCHLUCKER (Gegen 404 Fehler)
+# 3. MÜLLSCHLUCKER
 # ==========================================
 @app.post("/vapi-incoming")
 async def dummy_incoming(request: Request):
-    return {"status": "ok"}
+    return {"status": "ok"}
+
+@app.get("/")
+def home():
+    return {"status": "Online", "docs_loaded": len(KNOWLEDGE_CACHE)}