Upload main.py

main.py

@@ -10,6 +10,7 @@ app = FastAPI()
 # --- SETUP ---
 COLLECTION_KNOWLEDGE = "knowledge_base"
 COLLECTION_RULES = "availability_rules"
+COLLECTION_INBOX = "inbox"  # NEU: Hier landen ungelöste Fragen
 KNOWLEDGE_CACHE = []
 
 # --- FIREBASE VERBINDUNG ---
@@ -49,53 +50,39 @@ def get_stem(word):
     return w
 
 # --- HELPER: VAPI REQUEST PARSER ---
-# Holt sicher die ID und die Argumente aus dem Vapi-Salat
 def parse_vapi_request(data):
     tool_call_id = "unknown"
     args = {}
-    
     try:
         msg = data.get("message", {})
-        
-        # Variante A: Dein Log-Format ("toolCallList")
         if "toolCallList" in msg:
             call = msg["toolCallList"][0]
             tool_call_id = call["id"]
             if "function" in call and "arguments" in call["function"]:
                 args = call["function"]["arguments"]
-        
-        # Variante B: Standard Vapi ("toolCalls")
         elif "toolCalls" in msg:
             call = msg["toolCalls"][0]
             tool_call_id = call["id"]
             if "function" in call and "arguments" in call["function"]:
                 args = call["function"]["arguments"]
-                
-        # Argumente sind oft ein String, müssen zu JSON werden
         if isinstance(args, str):
             args = json.loads(args)
-            
     except Exception as e:
         print(f"⚠️ Parsing Info: {e}")
-        
     return tool_call_id, args
 
 # ==========================================
-# TOOL 1: VERFÜGBARKEIT (Strict Vapi Format)
+# TOOL 1: VERFÜGBARKEIT
 # ==========================================
 @app.post("/check_availability")
 async def check_availability(request: Request):
     data = await request.json()
-    print(f"🚦 TOOL REQUEST: {json.dumps(data)[:100]}...") # Debug Log
-
-    # 1. ID extrahieren (WICHTIG!)
     tool_call_id, _ = parse_vapi_request(data)
     
     today = datetime.now().strftime("%Y-%m-%d")
     status = "available"
     instruction = "Normal arbeiten"
     
-    # 2. Datenbank Check
     try:
         if db:
             rules = db.collection(COLLECTION_RULES).where("active", "==", True).stream()
@@ -103,48 +90,40 @@ async def check_availability(request: Request):
                 rd = r.to_dict()
                 if rd.get('start_date') <= today <= rd.get('end_date'):
                     print(f"🛑 REGEL AKTIV: {rd.get('name')}")
-                    status = "unavailable"
+                    # Einfache Logik: Wenn "Ferien" im Namen -> Limited, sonst Unavailable
+                    if "ferien" in rd.get('name', '').lower():
+                        status = "limited"
+                    else:
+                        status = "unavailable"
                     instruction = rd.get('instruction_text')
                     break
     except Exception as e:
         print(f"❌ ERROR CHECK: {e}")
 
-    print(f"👉 RESULTAT: {status} | {instruction}")
-    
-    # 3. Antwort im strikten Vapi-Format
     return {
-        "results": [
-            {
-                "toolCallId": tool_call_id,
-                "result": {
-                    "status": status,
-                    "instruction": instruction
-                }
-            }
-        ]
+        "results": [{"toolCallId": tool_call_id, "result": {"status": status, "instruction": instruction}}]
     }
 
 # ==========================================
-# TOOL 2: SUCHE (Jetzt auch mit ID Rückgabe!)
+# TOOL 2: SUCHE (Mit Inbox-Speicherung!)
 # ==========================================
 @app.post("/search")
 async def search(request: Request):
     data = await request.json()
-    
-    # 1. ID und Query extrahieren
     tool_call_id, args = parse_vapi_request(data)
     query = args.get("search_query") or args.get("query") or data.get("search_query")
     
     print(f"🔎 FRAGE (ID: {tool_call_id}): '{query}'")
     
-    answer_text = "Dazu habe ich keine Informationen."
+    answer_text = "Dazu habe ich leider keine Informationen in meiner Datenbank."
     
     if query:
-        # --- SUCHE LOGIK (Mit Udo/Capaneo Boost) ---
         STOP_WORDS = ["hallo", "guten", "tag", "moin", "bitte", "danke", "frage"]
         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]
         
+        found = False
+        
         if relevant_words:
             best_doc = None
             best_score = 0
@@ -164,30 +143,31 @@ async def search(request: Request):
                     best_score = score
                     best_doc = doc
 
+            # SCHWELLE: 20 PUNKTE
             if best_doc and best_score >= 20:
                 print(f"🏆 TREFFER ({best_score}): {best_doc.get('question')}")
                 answer_text = best_doc.get("answer")
+                found = True
             else:
                  print(f"⚠️ Zu wenig Relevanz (Max: {best_score})")
+        
+        # --- NEU: SPEICHERN WENN NICHT GEFUNDEN ---
+        if not found and db:
+            print("📥 Speichere in Inbox...")
+            db.collection(COLLECTION_INBOX).add({
+                "query": query,
+                "timestamp": datetime.now(),
+                "status": "open"
+            })
 
-    # 3. Antwort im strikten Vapi-Format
-    # Bei 'search' erwartet das LLM meist einfach den Text im result
     return {
-        "results": [
-            {
-                "toolCallId": tool_call_id,
-                "result": answer_text 
-            }
-        ]
+        "results": [{"toolCallId": tool_call_id, "result": answer_text}]
     }
 
-# ==========================================
-# 3. MÜLLSCHLUCKER
-# ==========================================
 @app.post("/vapi-incoming")
 async def dummy_incoming(request: Request):
     return {"status": "ok"}
 
 @app.get("/")
 def home():
-    return {"status": "Online", "docs_loaded": len(KNOWLEDGE_CACHE)}
+    return {"status": "Online", "docs": len(KNOWLEDGE_CACHE)}