Update app.py

app.py

@@ -41,9 +41,14 @@ def response(prompt, history, selected_option, temperature=0.9, max_new_tokens=5
         do_sample=True,
         seed=42,
     )
+    if selected_option == "Keine These ausgewählt":
+        selected_option = " "
+    else:
+        selected_option = "Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + selected_option
+    queryprompt= prompt + " "+ selected_option
     addon = ""
     results = collection.query(
-        query_texts=[prompt],
+        query_texts=[queryprompt],
         n_results=60,
     )
     dists = ["<br><small>(relevance: " + str(round((1 - d) * 100) / 100) + ";" for d in results['distances'][0]]
@@ -53,10 +58,7 @@ def response(prompt, history, selected_option, temperature=0.9, max_new_tokens=5
     if len(results) > 1:
         addon = "Bitte berücksichtige bei deiner Antwort ausschießlich 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)
     system = "Du heißt Avah und bist ein deutschsprachiges KI-basiertes Wahlhilfe Beratungssystem zum Wahl-O-Maten für die Bundestagswahl 2021. Du nimmst selber keine eigene politische Postion ein!" + addon + "\n\nUser-Anliegen:"
-    if selected_option == "Keine These ausgewählt":
-        selected_option = " "
-    else:
-        selected_option = "Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + selected_option
+    
     formatted_prompt = format_prompt(system + "\n" + selected_option + "\n" + prompt, history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""