Update app.py

app.py

@@ -3,13 +3,14 @@ import os
 import gradio as gr
 import json
 from huggingface_hub import InferenceClient
+import pandas as pd
 
 path = '/Users/thiloid/Desktop/Avah/chromafiles'
 if not os.path.exists(path): 
     path = "/home/user/app/chromafiles"
 
 print(path)
-
+mragdata= pd.read_excel("mragdata.xlsx")
 client = chromadb.PersistentClient(path=path)
 print(client.heartbeat()) 
 print(client.get_version())  
@@ -157,13 +158,15 @@ def responsefull(
         #return "Bitte spezifiziere deine Frage."
     if responsecritical(prompt)==False:
         return "Es scheint so, als sei dies keine Frage, die sich auf Bundestagswahl oder den Wahl-O-Maten bezieht"
-    
+    arag=False
     if selected_option == "Keine These ausgewählt":
+        arag=True
         query_prompt=format_prompt(prompt,history)
         selected_optionfull = " Die Frage bezieht sich auf keine These speziell und ist somit allgemeiner zu behandeln."
     else:
-        query_prompt=selected_option
-        selected_optionfull = " Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + selected_option
+        results=mpragdata[mpragdata["These: These"] == selected_option]["testnew"]
+        #query_prompt=selected_option
+        #selected_optionfull = " Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + selected_option
     if contains_op(prompt)==True:
         return "Als KI Assitenz kann ich dich nur beraten, jedoch bleibe ich neutral und werde keine Position einehmen. Frage mich aber gerne etwas anderes zum Wahl-O-Maten."
     #time.sleep(10)
@@ -172,21 +175,22 @@ def responsefull(
     
     prompt= prompt + selected_optionfull
     addon=""
-    results=collection.query(
+    if arag:
+        results=collection.query(
       query_texts=[query_prompt],
       n_results=8,
       #where={"source": "google-docs"}
       #where_document={"$contains":"search_string"}
     )
     #print("REsults")
-    print(results)
+        print(results)
     #print("_____")
-    dists=["<br><small>(relevance: "+str(round((1-d)*100)/100)+";" for d in results['distances'][0]]
+        dists=["<br><small>(relevance: "+str(round((1-d)*100)/100)+";" for d in results['distances'][0]]
   
     #sources=["source: "+s["source"]+")</small>" for s in results['metadatas'][0]]
-    results=results['documents'][0]
-    combination = zip(results,dists)
-    combination = [' '.join(triplets) for triplets in combination]
+        results=results['documents'][0]
+        combination = zip(results,dists)
+        combination = [' '.join(triplets) for triplets in combination]
     #print(str(prompt)+"\n\n"+str(combination))
     if(len(results)>1):
         addon="Bitte berücksichtige bei deiner Antwort folgende Auszüge aus unserer Wahlhilfe Datenbank, sofern sie für die Antwort erforderlich sind. Beantworte die Frage knapp und präzise. Ignoriere unpassende Datenbank-Auszüge OHNE sie zu kommentieren, zu erwähnen oder aufzulisten:\n"+"\n".join(results)