Add augmentation-option

run.py

@@ -66,7 +66,7 @@ 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 Blogbeitrag zur Anfrage des Users!"},
+      {"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!"}], 
     ids=[str(len(x)+1),str(len(x)+2),str(len(x)+3)] 
@@ -212,7 +212,7 @@ from datetime import datetime
 import os
 import re
 
-def response(message, history,customSysPrompt, genre, hfToken):
+def response(message, history,customSysPrompt, genre, hfToken, augmentation):
   if((onPrem==False) & (hfToken.startswith("hf_"))): # use HF-hub with custom token if token is provided
     from huggingface_hub import InferenceClient
     import gradio as gr
@@ -224,6 +224,7 @@ def response(message, history,customSysPrompt, genre, hfToken):
   removeHTML=True
   system=customSysPrompt # system-prompt can be changed in the UI (usually defaults to something like the following system-prompt)
   if(system==""): system="Du bist wissenschaftlicher Mitarbeiter an einem Forschungsinstitut und zuständig für die Wissenschaftskommunikation."
+  if(augmentation==True): system=system+"\nFür eine besonders gelungene Lösung erhältst du eine Gehaltserhöhung! Schreibe deine Texte in natürlicher und einfacher Sprache. Zielgruppe sind deutschsprachige Personen mit unterschiedlichen Bildungshintergründen."
   message=message.replace("[INST]","")
   message=message.replace("[/INST]","")
   message=message.replace("</s>","")
@@ -335,7 +336,8 @@ gr.ChatInterface(
       ["Blogbeitrag","Gliederungsvorschlag","Social Media Beitrag",""],
       value="Blogbeitrag",
       label="Genre"),
-     gr.Textbox(
+    gr.Checkbox(label="Motivationsschub", info="Ergänze System Prompt um weitere Anregungen für pot. hochwertigere Ergebnisse."),
+    gr.Textbox(
       value="",
       label="HF_token"),     
   ]