Update app.py

app.py

@@ -263,8 +263,9 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
             splittet = False
 
         #kein Bild hochgeladen -> auf Text antworten...
+        status = "Antwort der KI ..."
         if (file == None):
-            result = generate_text(prompt, chatbot, history, rag_option, model_option, openai_api_key, db, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,)
+            result, status = generate_text(prompt, chatbot, history, rag_option, model_option, openai_api_key, db, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,)
             history = history + [(prompt, result)]
         else:
             #Es wurde ein Bild angehängt -> wenn prompt dazu, das Bild analysieren
@@ -281,7 +282,7 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         for character in result:
             chatbot[-1][1] += character
             time.sleep(0.03)
-            yield chatbot, history, None, "Generating"
+            yield chatbot, history, None, status
             if shared_state.interrupted:
                 shared_state.recover()
                 try:
@@ -311,7 +312,7 @@ 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'))
         chatbot[-1][1] = "<img src='data:image/png;base64,{0}'/>".format(response.data[0].b64_json)
     
-    return chatbot, "Success"
+    return chatbot, "Antwort KI: Success"
     
 
 ##################################################
@@ -362,6 +363,7 @@ def generate_text_zu_doc(file, prompt, k, rag_option, chatbot, db):
 #mit oder ohne RAG möglich
 def generate_text (prompt, chatbot, history, rag_option, model_option, openai_api_key, db, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,):
     global splittet
+    suche_im_Netz="Antwort der KI ..."
     print("Text pur..............................")
     if (openai_api_key == "" or openai_api_key == "sk-"):
         #raise gr.Error("OpenAI API Key is required.")
@@ -417,15 +419,16 @@ def generate_text (prompt, chatbot, history, rag_option, model_option, openai_ap
             print("LLM aufrufen ohne RAG: ...........")
             result = llm_chain(llm, history_text_und_prompt)
 
-        #Wenn keine Antwort möglich "Ich weiß es nicht", dann versuchen mit Suche im Internet.
-        if (result == "Ich weiß es nicht."):
+        #Wenn keine Antwort möglich "Ich weiß es nicht" etc., dann versuchen mit Suche im Internet.
+        if is_response_similar(result):
             print("Suche im Netz: ...........")
+            suche_im_Netz="Antwort aus dem Internet ..."
             result = create_assistant_suche(history_text_und_prompt)
         
     except Exception as e:
         raise gr.Error(e)
 
-    return result
+    return result, suche_im_Netz
 
         
       
@@ -472,7 +475,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     with gr.Tab("LI Chatbot"):
         #with gr.Row():
             #gr.HTML("LI Chatot")
-            #status_display = gr.Markdown("Success", visible = False, elem_id="status_display")
+            status_display = gr.Markdown("Antwort der KI ...", visible = False, elem_id="status_display")
         with gr.Row():
             with gr.Column(scale=5):
                 with gr.Row():
@@ -546,7 +549,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     with gr.Tab("LI Zeichnen"):
         #with gr.Row():
             #gr.HTML("LI Zeichnen mit KI")
-            #status_display2 = gr.Markdown("Success", visible = False, elem_id="status_display")
+            status_display2 = gr.Markdown("Success", visible = False, elem_id="status_display")
         #gr.Markdown(description2)
         with gr.Row():
             with gr.Column(scale=5):