Update app.py

app.py

@@ -31,7 +31,7 @@ def format_prompt(message, history):
   return prompt
 
 def response(
-    prompt, history,temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
+    prompt, history,selected_option,temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
 ):
     temperature = float(temperature)
     if temperature < 1e-2: temperature = 1e-2
@@ -66,8 +66,8 @@ def response(
     #print(str(prompt)+"\n\n"+str(combination))
     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."+addon+"\n\nUser-Anliegen:"   
-    formatted_prompt = format_prompt(system+"\n"+prompt,history)
+    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 in keinem Fall eigen politische Postionen 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)
     output = ""
     for response in stream:
@@ -79,6 +79,8 @@ def response(
 def update_prompt(selected_option, history):
     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
     return history, selected_option
 
 dropdown_options = ["Keine These ausgewählt",