Update app.py

app.py

@@ -10,7 +10,6 @@ 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())  
@@ -158,15 +157,11 @@ 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:
-        results=str(mragdata[mragdata["These: These"] == selected_option]["testnew"])
-        print(results)
-        #query_prompt=selected_option
+        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."
@@ -176,22 +171,21 @@ def responsefull(
     
     prompt= prompt + selected_optionfull
     addon=""
-    if arag:
-        results=collection.query(
+    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)