Update app.py

app.py

@@ -28,9 +28,9 @@ creds = service_account.Credentials.from_service_account_info(json_data, scopes=
 client = gspread.authorize(creds)
 sheet = client.open("Avah Eval").sheet1  # Open the sheet
 
-def save_to_sheet(date, name, message, IP, dev, header):
+def save_to_sheet(date, name, message, IP, dev, header,IDval):
     # Write user input to the Google Sheet
-    sheet.append_row([date, name, message, IP, dev, header])
+    sheet.append_row([date, name, message, IP, dev, header,IDval])
     return f"Thanks {name}, your message has been saved!"
 
 
@@ -166,7 +166,10 @@ def is_single_word_question(question):
     # Frage in Wörter aufteilen
     words = question.split()
     # Überprüfen, ob nur ein Wort vorhanden ist
-    return len(words) == 1
+    if len(words) == 1:
+        if words.lower()=="fertig":
+            return 1
+    return 0
 def get_value_after_question_mark(url):
     url=str(url)
     # Find the position of the question mark
@@ -220,12 +223,13 @@ def responsefull(
     print(full_url)
     headers = request.headers
     IP, dev = extract_ip_and_device(headers)
-    
+    if is_single_word_question(prompt)==1:
+        return "Hier ist der Link zur Nachbefragung: https://tudaipol.limesurvey.net/488161?uid="+ str(IDval)
     #if is_single_word_question(prompt)==1:
         #return "Bitte spezifiziere deine Frage."
     if responsecritical(prompt)==False:
         now = str(datetime.now())
-        save_to_sheet(now, prompt, "Critical", IP, dev, str(headers))
+        save_to_sheet(now, prompt, "Critical", IP, dev, str(headers),IDval)
         return "Es scheint so, als sei dies keine Frage, die sich auf Bundestagswahl oder den Wahl-O-Maten bezieht"
     if selected_option == "Keine These ausgewählt":
         query_prompt=format_prompt(prompt,history)
@@ -235,12 +239,12 @@ def responsefull(
         selected_optionfull = " Die Nutzer Frage bezieht sich auf folgende Wahl-O-Mat These: " + selected_option
     if contains_op(prompt)==True:
         now = str(datetime.now())
-        save_to_sheet(now, prompt, "oppinion", IP, dev, str(headers))
+        save_to_sheet(now, prompt, "oppinion", IP, dev, str(headers),IDval)
         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."
     #time.sleep(10)
     if responsevote(prompt)==True:
         now = str(datetime.now())
-        save_to_sheet(now, prompt, "vote", IP, dev, str(headers))
+        save_to_sheet(now, prompt, "vote", IP, dev, str(headers),IDval)
         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."
     
     prompt= prompt + selected_optionfull
@@ -270,7 +274,7 @@ def responsefull(
     for response in stream:
         output += response.token.text
     now = str(datetime.now())
-    save_to_sheet(now, prompt, output, IP, dev, str(headers))
+    save_to_sheet(now, prompt, output, IP, dev, str(headers),IDval)
     return output