Update run.py

run.py

@@ -18,9 +18,8 @@ print(client.list_collections())
 from chromadb.utils import embedding_functions
 default_ef = embedding_functions.DefaultEmbeddingFunction()
 sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer")#"VAGOsolutions/SauerkrautLM-Mixtral-8x7B-Instruct")
-#instructor_ef = embedding_functions.InstructorEmbeddingFunction(model_name="hkunlp/instructor-large", device="cuda")
-#print(str(client.list_collections()))
-collection = client.get_collection(name="chromaTS", embedding_function=sentence_transformer_ef)
+
+collection = client.get_collection(name="chromatsc", embedding_function=sentence_transformer_ef)
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
@@ -68,8 +67,8 @@ def response(
     combination = [' '.join(triplets) for triplets in combination]
     #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 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 bist ein deutschsprachiges KI-basiertes Studienberater Assistenzsystem, das zu jedem Anliegen möglichst geeignete Studieninformationen empfiehlt."+addon+"\n\nUser-Anliegen:"   
+      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)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
@@ -78,8 +77,59 @@ def response(
         yield output
     #output=output+"\n\n<br><details open><summary><strong>Sources</strong></summary><br><ul>"+ "".join(["<li>" + s + "</li>" for s in combination])+"</ul></details>"
     yield output
+    
+def update_prompt(selected_option, history):
+    if selected_option=="Keine These ausgewählt":
+        selected_option== " "
+    return history, selected_option
+
+dropdown_options = ["Keine These ausgewählt", 
+                    'Auf allen Autobahnen soll ein generelles Tempolimit gelten.',
+ 'Deutschland soll seine Verteidigungsausgaben erhöhen.',
+ 'Bei Bundestagswahlen sollen auch Jugendliche ab 16 Jahren wählen dürfen.',
+ 'Die Förderung von Windenergie soll beendet werden.',
+ 'Die Möglichkeiten der Vermieterinnen und Vermieter, Wohnungsmieten zu erhöhen, sollen gesetzlich stärker begrenzt werden.',
+ 'Impfstoffe gegen Covid-19 sollen weiterhin durch Patente geschützt sein.',
+ 'Der für das Jahr 2038 geplante Ausstieg aus der Kohleverstromung soll vorgezogen werden.',
+ 'Alle Erwerbstätigen sollen in der gesetzlichen Rentenversicherung versichert sein müssen.',
+ 'Das Recht anerkannter Flüchtlinge auf Familiennachzug soll abgeschafft werden.',
+ 'Auf den Umsatz, der in Deutschland mit digitalen Dienstleistungen erzielt wird, soll eine nationale Steuer erhoben werden.',
+ 'Die traditionelle Familie aus Vater, Mutter und Kindern soll stärker als andere Lebensgemeinschaften gefördert werden.',
+ 'Spenden von Unternehmen an Parteien sollen weiterhin erlaubt sein.',
+ 'Studentinnen und Studenten sollen BAföG unabhängig vom Einkommen ihrer Eltern erhalten.',
+ 'In Deutschland soll es generell möglich sein, neben der deutschen eine zweite Staatsbürgerschaft zu haben.',
+ 'Bundesbehörden sollen in ihren Veröffentlichungen unterschiedliche Geschlechtsidentitäten sprachlich berücksichtigen.',
+ 'Die Ostsee-Pipeline "Nord Stream 2", die Gas von Russland nach Deutschland transportiert, soll wie geplant in Betrieb gehen dürfen.',
+ 'Der Solidaritätszuschlag soll vollständig abgeschafft werden.',
+ 'Das Tragen eines Kopftuchs soll Beamtinnen im Dienst generell erlaubt sein.',
+ 'Die Zulassung von neuen Autos mit Verbrennungsmotor soll auch langfristig möglich sein.',
+ 'Der Bund soll mehr Zuständigkeiten in der Schulpolitik erhalten.',
+ 'Der Bund soll Projekte zur Bekämpfung des Antisemitismus stärker finanziell unterstützen.',
+ 'Chinesische Firmen sollen keine Aufträge für den Ausbau der Kommunikationsinfrastruktur in Deutschland erhalten dürfen.',
+ 'Der Staat soll weiterhin für Religionsgemeinschaften die Kirchensteuer einziehen.',
+ 'Der kontrollierte Verkauf von Cannabis soll generell erlaubt sein.',
+ 'Deutschland soll aus der Europäischen Union austreten.',
+ 'Die Landeslisten der Parteien für die Wahlen zum Deutschen Bundestag sollen abwechselnd mit Frauen und Männern besetzt werden m��ssen.',
+ 'Stationäre Behandlungen im Krankenhaus sollen weiterhin über eine Fallpauschale abgerechnet werden.',
+ 'Auf hohe Vermögen soll wieder eine Steuer erhoben werden.',
+ 'Bei der Videoüberwachung öffentlicher Plätze soll Gesichtserkennungssoftware eingesetzt werden dürfen.',
+ 'Auch Ehepaare ohne Kinder sollen weiterhin steuerlich begünstigt werden.',
+ 'Ökologische Landwirtschaft soll stärker gefördert werden als konventionelle Landwirtschaft.',
+ 'Islamische Verbände sollen als Religionsgemeinschaften staatlich anerkannt werden können.',
+ 'Der staatlich festgelegte Preis für den Ausstoß von CO2 beim Heizen und Autofahren soll stärker steigen als geplant.',
+ 'Die Schuldenbremse im Grundgesetz soll beibehalten werden.',
+ 'Asyl soll weiterhin nur politisch Verfolgten gewährt werden.',
+ 'Der gesetzliche Mindestlohn soll spätestens im Jahr 2022 auf mindestens 12 Euro erhöht werden.',
+ 'Der Flugverkehr soll höher besteuert werden.',
+ 'Unternehmen sollen selbst entscheiden, ob sie ihren Beschäftigten das Arbeiten im Homeoffice erlauben.']
+                   ]
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(value=[[None,"Herzlich willkommen! Ich bin Chätti ein KI-basiertes Studienassistenzsystem, das für jede Anfrage die am besten Studieninformationen empfiehlt.<br>Erzähle mir, was du gerne tust!"]], render_markdown=True)
+    dropdown = gr.Dropdown(choices=dropdown_options, label="Choose an option")
+    dropdown.change(fn=update_prompt, inputs=[dropdown, chatbot], outputs=[chatbot, dropdown])
+    gr.ChatInterface(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."]],response, inputs=[gr.Textbox(), chatbot, dropdown], title="Avah (Artificial Voting Advice Helper)").queue().launch(share=True)
 
-gr.ChatInterface(response, chatbot=gr.Chatbot(value=[[None,"Herzlich willkommen! Ich bin Chätti ein KI-basiertes Studienassistenzsystem, das für jede Anfrage die am besten Studieninformationen empfiehlt.<br>Erzähle mir, was du gerne tust!"]],render_markdown=True),title="German Studyhelper Chätti").queue().launch(share=True) #False, server_name="0.0.0.0", server_port=7864)
 print("Interface up and running!")