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,IDval):
+def save_to_sheet(date, name, message, IP, dev, header):
     # Write user input to the Google Sheet
-    sheet.append_row([date, name, message, IP, dev, header,IDval])
+    sheet.append_row([date, name, message, IP, dev, header])
     return f"Thanks {name}, your message has been saved!"
 
 
@@ -50,7 +50,7 @@ sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFuncti
 
 collection = client.get_collection(name="chromatsc", embedding_function=sentence_transformer_ef)
 
-client = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407")#"mistralai/Mixtral-8x7B-Instruct-v0.1")#"mistralai/Mistral-Nemo-Instruct-2407")
+client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def format_promptc(message):
   prompt = "" #"<s>"
@@ -100,7 +100,6 @@ def responsevote(
     output = ""
     for response in stream:
         output += response.token.text
-    print("VOTE")
     print(output)
     sentence_lower = output.lower()
     # Check if the word 'nein' is in the sentence
@@ -141,8 +140,7 @@ def responsecritical(
     output = ""
     for response in stream:
         output += response.token.text
-    print("CRITICAL")
-    print(output)
+    #print(output)
     sentence_lower = output.lower()
     # Check if the word 'nein' is in the sentence
     if 'ja' in sentence_lower:
@@ -168,30 +166,19 @@ def is_single_word_question(question):
     # Frage in Wörter aufteilen
     words = question.split()
     # Überprüfen, ob nur ein Wort vorhanden ist
-    if len(words) == 1:
-        if words[0].lower() == "fertig":  # `words[0]` refers to the single word in the list
-            return 1
-    return 0
+    return len(words) == 1
 def get_value_after_question_mark(url):
-    url = str(url)
+    url=str(url)
     # Find the position of the question mark
     question_mark_index = url.find('?')
     
     # Extract the part after the question mark
     if question_mark_index != -1:
-        # Get the part after the question mark
-        query_string = url[question_mark_index + 1:]
-        
-        # Split the query string into key-value pairs
-        pairs = query_string.split('&')
-        
-        # Iterate over the key-value pairs
-        for pair in pairs:
-            key, value = pair.split('=')
-            if key == 'var1':  # Replace 'var1' with the key you are interested in
-                return value
+        value_after_question_mark = url[question_mark_index + 1:].split('=')[0]
+    else:
+        value_after_question_mark = ""
     
-    return ""
+    return value_after_question_mark
 def extract_ip_and_device(headers_obj):
     ip_address = None
     device_info = None
@@ -230,18 +217,15 @@ def responsefull(
     #get ID
     full_url = request.url
     IDval= get_value_after_question_mark(full_url)
-    IDval=str(IDval)
-    print("URL")
     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),IDval)
+        save_to_sheet(now, prompt, "Critical", IP, dev, str(headers))
         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)
@@ -251,12 +235,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),IDval)
+        save_to_sheet(now, prompt, "oppinion", IP, dev, str(headers))
         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),IDval)
+        save_to_sheet(now, prompt, "vote", IP, dev, str(headers))
         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
@@ -286,7 +270,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),IDval)
+    save_to_sheet(now, prompt, output, IP, dev, str(headers))
     return output
 
 
@@ -331,7 +315,7 @@ dropdown_options = ["Keine These ausgewählt",
  'Unternehmen sollen selbst entscheiden, ob sie ihren Beschäftigten das Arbeiten im Homeoffice erlauben.']
 
 with gr.Blocks(title="test") as demo:
-    chatbot = gr.Chatbot(value=[[None, "Hallo mein Name ist AVAH (Artificial Voting Advice Helper), deine interaktive Hilfe zum Wahl-O-Maten für die Bundestagswahl 2021. Wenn du Fragen zu Thesen der Wahlhilfe hast, wähle gerne die besagt These aus. Ansonsten kannst du mir auch unabhängig von bestimmten Thesen Fragen stellen. Sobald du fertig mit den Fragen bist, schreibe bitte `fertig` in den Chat um zur Nachbefragung zu gelangen."]], render_markdown=True)
+    chatbot = gr.Chatbot(value=[[None, "Hallo mein Name ist AVAH (Artificial Voting Advice Helper), deine interaktive Hilfe zum Wahl-O-Maten für die Bundestagswahl 2021. Wenn du Fragen zu Thesen der Wahlhilfe hast, wähle gerne die besagt These aus. Ansonsten kannst du mir auch unabhängig von bestimmten Thesen Fragen stellen."]], render_markdown=True)
     dropdown = gr.Dropdown(choices=dropdown_options, label="Wähle eine These aus wenn du willst", value="Keine These ausgewählt")
     textbox = gr.Textbox(label="Deine Frage")