Update run.py

run.py

@@ -2,7 +2,7 @@
 # Title:  Gradio Writing Assistant
 # Author: Andreas Fischer
 # Date:   May 23th, 2024
-# Last update:June 7th, 2024
+# Last update:June 12th, 2024
 ##########################################################################################
 
 #https://github.com/abetlen/llama-cpp-python/issues/306
@@ -66,9 +66,9 @@ if(len(x)==0):
   collection.add(
     documents=["Ich möchte einen Blogbeitrag","Ich möchte einen Gliederungsvorschlag","Ich möchte einen Social Media Beitrag"], 
     metadatas=[
-      {"prompt": "Bitte schreibe einen detaillierten Blogbeitrag zur Anfrage des Users, mit allen relevanten Informationen zum Thema!"},
-      {"prompt": "Bitte entwerfe einen Gliederungsvorschlag zur Anfrage des Users!"},
-      {"prompt": "Bitte verfasse einen Beitrag für die professionelle social media Plattform LinkedIn zur Anfrage des Users!"}], 
+      {"prompt": "Bitte schreibe einen detaillierten Blogbeitrag zur Anfrage des Users, mit allen relevanten Informationen zum Thema!", "genre":"Beitrag"},
+      {"prompt": "Bitte entwerfe einen Gliederungsvorschlag zur Anfrage des Users!", "genre":"Gliederungsvorschlag"},
+      {"prompt": "Bitte verfasse einen Beitrag für die professionelle social media Plattform LinkedIn zur Anfrage des Users!", "genre":"Social Media Beitrag"}], 
     ids=[str(len(x)+1),str(len(x)+2),str(len(x)+3)] 
   )
 
@@ -236,7 +236,7 @@ def response(message, history,customSysPrompt, genre, augmentation, hfToken):
     RAGResults=collection.query(query_texts=[message], n_results=1)
     genre=str(RAGResults['documents'][0][0]) # determine genre based on best-matching db-entry 
   
-  rag="\n\n"+collection.get(where_document={"$contains":genre},include=["metadatas"])['metadatas'][0]['prompt'] # genre-specific addendum to system prompt (rag)
+  rag="\n\n"+collection.get(where={"genre": genre},include=["metadatas"])['metadatas'][0]['prompt'] # genre-specific addendum to system prompt (rag)
   if(len(history)>0):
     rag=rag+"\nFalls der User Rückfragen oder Änderungsvorschläge zu deinem Entwurf hat, gehe darauf ein." # add dialog-specific addendum to rag
   
@@ -329,15 +329,16 @@ gr.ChatInterface(
   chatbot=gr.Chatbot(value=val, render_markdown=True),
   title="KI Schreibassistenz (lokal)" if onPrem else "KI Schreibassistenz",
   additional_inputs=[
-    gr.Textbox(
+    gr.Textbox(info="Basiskomponente der Anweisungen, die vor dem Dialog an das System gehen.",
       value="Du bist wissenschaftlicher Mitarbeiter an einem Forschungsinstitut und zuständig für die Wissenschaftskommunikation.",
       label="System Prompt"),
-    gr.Dropdown(
-      ["Blogbeitrag","Gliederungsvorschlag","Social Media Beitrag",""],
-      value="Blogbeitrag",
+    gr.Dropdown(info="Wähle das gewünschte Genre des zu schreibenden Textes",
+      ["Beitrag","Gliederungsvorschlag","Social Media Beitrag",""],
+      value="Beitrag",
       label="Genre"),
-    gr.Checkbox(label="Motivationsschub", info="Ergänze System Prompt um weitere Anregungen für pot. hochwertigere Ergebnisse."),
-    gr.Textbox(
+    gr.Checkbox(info="Optional: Automatische Ergänzung des System Prompt um Formulierungen, die hochwertigere Ergebnisse erwarten lassen.",
+      label="Motivationsschub"),
+    gr.Textbox(info="Optional: Gib einen gültigen Huggingface Access Token an, um mehr Texte produzieren zu können.",
       value="",
       label="HF_token"),     
   ]