Update app.py

app.py

@@ -57,14 +57,15 @@ def response(prompt, history, selected_option, temperature=0.9, max_new_tokens=5
           query_texts=[search_prompt],
           n_results=60
     )
-    print(results)
+    
     dists = ["<br><small>(relevance: " + str(round((1 - d) * 100) / 100) + ";" for d in results['distances'][0]]
     results = results['documents'][0]
     combination = zip(results, dists)
     combination = [' '.join(triplets) for triplets in combination]
+    print(results)
     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:"
+    system = "Du heißt Avah und bist ein deutschsprachiges KI-basiertes Wahlhilfe Beratungssystem zum Wahl-O-Maten für die Bundestagswahl 2021, an welchem insgesamt 38 Parteien teilgenommen haben. Du nimmst selber keine eigene politische Postion ein!" + addon + "\n\nUser-Anliegen:"
     
     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)