Update app.py

app.py

@@ -230,6 +230,7 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         global splittet
         #wenn RAG angeschaltet - Vektorstore initialisieren
         #aber nur, wenn es noch nicht geshehen ist (splittet = False)
+        
         if (rag_option == "An"):
             #muss nur einmal ausgeführt werden... 
             if not splittet:
@@ -238,20 +239,21 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
             db = document_retrieval_chroma2()
             splittet = True
         else:
+            db=None
             splittet = False
 
         #kein Bild hochgeladen -> auf Text antworten...
         if (file == None):
-            result = generate_text(prompt, chatbot, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,)
+            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,)
             history = history + [(prompt, result)]
         else:
             #Es wurde ein Bild angehängt -> wenn prompt dazu, das Bild analysieren
             #geht nur über spezielle OpenAI-Schnittstelle...
             ext = analyze_file(file)
             if (ext == "png" or ext == "PNG" or ext == "jpg" or ext == "jpeg" or ext == "JPG" or ext == "JPEG"):
-                result= generate_text_zu_bild(file, prompt, k, rag_option, chatbot)
+                result= generate_text_zu_bild(file, prompt, k, rag_option, chatbot, db)
             else:
-                result = generate_text_zu_doc(file, prompt, k, rag_option, chatbot)
+                result = generate_text_zu_doc(file, prompt, k, rag_option, chatbot, db)
                 
             history = history + [((file,), None),(prompt, result)]
 
@@ -294,7 +296,7 @@ def generate_bild(prompt, chatbot, model_option_zeichnen='HuggingFace', temperat
 
 ##################################################
 #zu einem Bild und Text-Prompt eine Analyse generieren
-def generate_text_zu_bild(file, prompt, k, rag_option, chatbot):
+def generate_text_zu_bild(file, prompt, k, rag_option, chatbot, db):
     global splittet
     print("Text mit Bild ..............................")
     print(file)
@@ -318,7 +320,7 @@ def generate_text_zu_bild(file, prompt, k, rag_option, chatbot):
 
 ##################################################
 #zu einem Bild und Text-Prompt eine Analyse generieren
-def generate_text_zu_doc(file, prompt, k, rag_option, chatbot):
+def generate_text_zu_doc(file, prompt, k, rag_option, chatbot, db):
     global splittet
     print("text mit doc ..............................")
     
@@ -338,7 +340,7 @@ def generate_text_zu_doc(file, prompt, k, rag_option, chatbot):
 ####################################################
 #aus einem Text-Prompt die Antwort von KI bekommen
 #mit oder ohne RAG möglich
-def generate_text (prompt, chatbot, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,):
+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
     print("Text pur..............................")
     if (openai_api_key == "" or openai_api_key == "sk-"):