Thiloid commited on
Commit
0efd9c9
·
verified ·
1 Parent(s): 2669ff0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -117,7 +117,7 @@ def is_single_word_question(question):
117
  return len(words) == 1
118
 
119
  def responsefull(
120
- prompt, statemementval1, these1,history,temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
121
  ):
122
  temperature = float(temperature)
123
  if temperature < 1e-2: temperature = 1e-2
@@ -135,18 +135,19 @@ def responsefull(
135
  if responsecritical(prompt)==False:
136
  return "Es scheint so, als sei dies keine Frage, die sich auf Bundestagswahl oder den Wahl-O-Maten bezieht"
137
 
138
- if statemementval1 == 2:
139
- selected_option = "Die Frage bezieht sich auf keine These speziell und ist somit allgemeiner zu behandeln."
140
  else:
141
- selected_option = "Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + these1
142
  if contains_op(prompt)==True:
143
  return "Als KI Assitenz kann ich dich nur beraten, jedoch bleibe ich neutral und werde keine Position einehmen. Frage mich aber geren etwas anders zum Wahl-O-Maten."
144
  #time.sleep(10)
145
  if responsevote(prompt)==True:
146
  return "Als KI Assitenz bin ich politisch neutral. Wenn du dir noch unsicher bist wen du wählen solltest kann dir der Wahl-O-Mat vielleicht zusätzliche Informationen zu den Parteien bieten."
147
- prompt= prompt + selected_option
 
 
148
  addon=""
149
- query_prompt=format_prompt(prompt,history)
150
  results=collection.query(
151
  query_texts=[query_prompt],
152
  n_results=60,
 
117
  return len(words) == 1
118
 
119
  def responsefull(
120
+ prompt, history,selected_option,temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,
121
  ):
122
  temperature = float(temperature)
123
  if temperature < 1e-2: temperature = 1e-2
 
135
  if responsecritical(prompt)==False:
136
  return "Es scheint so, als sei dies keine Frage, die sich auf Bundestagswahl oder den Wahl-O-Maten bezieht"
137
 
138
+ if selected_option == "Keine These ausgewählt":
139
+ selected_optionfull = " Die Frage bezieht sich auf keine These speziell und ist somit allgemeiner zu behandeln."
140
  else:
141
+ selected_optionfull = " Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + selected_option
142
  if contains_op(prompt)==True:
143
  return "Als KI Assitenz kann ich dich nur beraten, jedoch bleibe ich neutral und werde keine Position einehmen. Frage mich aber geren etwas anders zum Wahl-O-Maten."
144
  #time.sleep(10)
145
  if responsevote(prompt)==True:
146
  return "Als KI Assitenz bin ich politisch neutral. Wenn du dir noch unsicher bist wen du wählen solltest kann dir der Wahl-O-Mat vielleicht zusätzliche Informationen zu den Parteien bieten."
147
+
148
+ query_prompt=format_prompt(prompt+ ". " + selected_option +,history)
149
+ prompt= prompt + selected_optionfull
150
  addon=""
 
151
  results=collection.query(
152
  query_texts=[query_prompt],
153
  n_results=60,