Update app.py

app.py

@@ -1,25 +1,34 @@
+import os
 import gradio as gr
 import requests
-import json
-import os
+
+# API Keys ophalen uit Hugging Face Secrets
+WEAVIATE_API_KEY = os.getenv("WEAVIATE_API_KEY")
+MISTRAL_API_KEY = os.getenv("MISTRAL_API_KEY")
 
 # Weaviate-instellingen
 WEAVIATE_URL = "https://ojk79flbtdqux0f4uua2qa.c0.europe-west3.gcp.weaviate.cloud/v1/graphql"
-WEAVIATE_API_KEY = os.getenv("WEAVIATE_APIKEY")
+HEADERS = {
+    "Authorization": f"Bearer {WEAVIATE_API_KEY}",
+    "X-Weaviate-Api-Key": WEAVIATE_API_KEY,
+    "Content-Type": "application/json"
+}
 
-if WEAVIATE_API_KEY:
-    print("✅ DEBUG: Weaviate API-key is succesvol geladen!")
-else:
-    print("❌ DEBUG: Weaviate API-key NIET gevonden! Controleer je Hugging Face Secrets.")
+# **Nieuwe consistente system message**
+SYSTEM_MESSAGE = (
+    "Je bent een formele, feitelijke en beknopte digitale assistent bij Logius. "
+    "Je helpt gebruikers met vragen over DigiD op basis van officiële documentatie. "
+    "Antwoorden zijn kort, bondig, feitelijk en eenvoudig te begrijpen."
+)
 
-# Functie om documentatie op te halen uit Weaviate
-def fetch_relevant_docs(query_text):
+# Functie om relevante documenten uit Weaviate op te halen
+def fetch_relevant_docs(user_query):
     query = {
         "query": f"""
         {{
           Get {{
             DigiD_Document_Vectorized(
-              nearText: {{ concepts: ["{query_text}"] }},
+              nearText: {{ concepts: ["{user_query}"] }},
               limit: 3
             ) {{
               title
@@ -30,45 +39,61 @@ def fetch_relevant_docs(query_text):
         }}
         """
     }
-    
-    headers = {
-    "Authorization": f"Bearer {WEAVIATE_API_KEY}",
-    "X-Weaviate-Api-Key": WEAVIATE_API_KEY,
-    "X-Weaviate-Cluster-Url": "https://ojk79flbtdqux0f4uua2qa.c0.europe-west3.gcp.weaviate.cloud",
-    "Content-Type": "application/json"
-    }
-    
-    response = requests.post(WEAVIATE_URL, json=query, headers=headers)
+
+    response = requests.post(WEAVIATE_URL, json=query, headers=HEADERS)
 
     if response.status_code == 200:
         data = response.json()
-        print("DEBUG: Volledige Weaviate API response:", json.dumps(data, indent=2))  # Volledige API-response printen
-        
         documents = data.get("data", {}).get("Get", {}).get("DigiD_Document_Vectorized", [])
+
         if not documents:
-            print("⚠️ Geen documenten gevonden voor de query:", query_text)
-        return documents
+            print(f"⚠️ Geen documenten gevonden voor de query: {user_query}")
+            return "Er is geen relevante documentatie gevonden voor deze vraag."
+
+        # Combineer de documentatie in een nette tekst
+        combined_text = "\n\n".join(
+            [f"**{doc['title']}**\n{doc['content']}\nBron: {doc['url']}" for doc in documents]
+        )
+
+        return combined_text
+
     else:
-        print(f"❌ Fout bij ophalen documentatie: {response.status_code}")
-        return []
-
-# Dummy-functie om chatbot te testen
-def respond(message, history):
-    print("\n📢 DEBUG: Gebruikersvraag ontvangen:", message)
-    docs = fetch_relevant_docs(message)
-    
-    if docs:
-        print("\n✅ DEBUG: Documenten opgehaald:")
-        for doc in docs:
-            print(f"- {doc['title']} ({doc['url']})")
-            print(f"  {doc['content'][:200]}...")  # Alleen eerste 200 tekens printen
+        print(f"❌ Fout bij ophalen documentatie: {response.status_code}, {response.text}")
+        return "Er is een fout opgetreden bij het ophalen van de documentatie."
+
+# Functie om antwoord te genereren met Mistral
+def generate_answer(user_query):
+    documentation = fetch_relevant_docs(user_query)
+
+    if "Er is geen relevante documentatie" in documentation:
+        return "Ik heb geen relevante documentatie kunnen vinden voor je vraag."
+
+    mistral_url = "https://api.mistral.ai/v1/chat/completions"
+    mistral_headers = {
+        "Authorization": f"Bearer {MISTRAL_API_KEY}",
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "model": "mistral-medium",
+        "messages": [
+            {"role": "system", "content": SYSTEM_MESSAGE},
+            {"role": "user", "content": user_query},
+            {"role": "assistant", "content": documentation}
+        ],
+        "temperature": 0.3,
+        "max_tokens": 512
+    }
+
+    response = requests.post(mistral_url, json=payload, headers=mistral_headers)
+
+    if response.status_code == 200:
+        return response.json()["choices"][0]["message"]["content"]
     else:
-        print("❌ DEBUG: Geen documenten opgehaald.")
-    
-    return "Check de logs voor opgehaalde documentatie!"
+        return f"Fout bij genereren antwoord: {response.status_code}, {response.text}"
 
-# Simpele Gradio UI
-demo = gr.ChatInterface(respond)
+# Gradio UI
+demo = gr.ChatInterface(generate_answer)
 
 if __name__ == "__main__":
     demo.launch()