Update app.py

app.py

@@ -97,6 +97,14 @@ API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffus
 ###############################################
 os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
 
+################################################
+#OpenAI Zugang, client und Assistant einmal erzeugen.
+################################################
+#zentral einmal erzeugen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
+client = OpenAI()
+assistant = client.beta.assistants.create(name="File Analysator",instructions=template, model="gpt-4-1106-preview",)
+thread = client.beta.threads.create()
+
 
 #################################################
 #################################################
@@ -135,8 +143,6 @@ def add_text2(chatbot, prompt):
 #nach dem Upload soll das zusätzliche Fenster mit dem image drinnen angezeigt werden
 def file_anzeigen(file):
     ext = analyze_file(file)
-    print("ext...............................")
-    print(ext)
     if (ext == "png" or ext == "PNG" or ext == "jpg" or ext == "jpeg" or ext == "JPG" or ext == "JPEG"):
         return gr.Image(width=47, visible=True, interactive = False,  height=47,  min_width=47,  show_label=False, show_share_button=False, show_download_button=False, scale = 0.5), file, file
     else:
@@ -202,11 +208,8 @@ def process_image(image_path, prompt):
 
 ##################################################
 #openassistant um uploaded Files zu analysieren
-def create_assistant(prompt, file): 
-    #zentral einmal erzeugen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
-    client = OpenAI()
-    assistant = client.beta.assistants.create(name="File Analysator",instructions=template, model="gpt-4-1106-preview",)
-    thread = client.beta.threads.create()
+def create_assistant(prompt, file):
+    #neues File dem Assistant hinzufügen
     file_neu = client.files.create(file=open(file,"rb",),purpose="assistants",)
     # Update Assistant
     assistant = client.beta.assistants.update(assistant.id,tools=[{"type": "code_interpreter"}, {"type": "retrieval"}],file_ids=[file_neu.id],)
@@ -266,7 +269,6 @@ def generate_bild(prompt, chatbot, model_option_zeichnen='HuggingFace', temperat
         chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
     else:     
         print("Bild Erzeugung DallE..............................")
-        client = OpenAI()
         #als Format ginge auch 'url', n - Anz. der erzeugten Bilder
         response = client.images.generate(model="dall-e-3",prompt=prompt,size="1024x1024",quality="standard",n=1, response_format='b64_json') 
         #chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
@@ -294,6 +296,9 @@ def generate_text_zu_bild(file, prompt, k, rag_option, chatbot):
         #prompt = generate_prompt_with_history_openai(neu_text_mit_chunks, history)
         #als reiner prompt:
         prompt_neu = generate_prompt_with_history(neu_text_mit_chunks, chatbot)
+        splittet = True
+    else:
+        splittet = False
 
     headers, payload = process_image(file, prompt_neu)
     response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=payload)
@@ -323,6 +328,9 @@ def generate_text_zu_doc(file, prompt, k, rag_option, chatbot):
         #prompt = generate_prompt_with_history_openai(neu_text_mit_chunks, history)
         #als reiner prompt:
         prompt_neu = generate_prompt_with_history(neu_text_mit_chunks, chatbot)
+        splittet = True
+    else:
+        splittet = False
         
     result = create_assistant(prompt_neu, file)
     return result