Update app.py

app.py

@@ -157,6 +157,13 @@ def umwandeln_fuer_anzeige(image):
         image.save(buffer, format='PNG')
         return buffer.getvalue()
 
+#########################################
+#Extension des hochgeladenen Files bestimmen
+def analyze_file(file):
+    file_extension = file.name.split('.')[-1]  # Holen Sie sich die Dateiendung
+    return file_extension
+
+
 ##########################################
 #ein hochgeladenes Bild so vorbereiten, dass OpenAI API es annehmen kann und bearbeiten
 #muss ein base64 Bils sein und header und payload entsprechend konfigurieren
@@ -208,9 +215,12 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         else:
             #Es wurde ein Bild angehängt -> wenn prompt dazu, das Bild analysieren
             #geht nur über spezielle OpenAI-Schnittstelle...
-            print("name.....................")
-            print(file.name)
-            result= "Ausgabe" #generate_text_zu_bild(file, prompt, k, rag_option, chatbot)
+            ext = analyze_file(file)
+            if (ext == "png" or ext == "PNG" or ext == "jgp" or ext == "jepg"):
+                result= "Ausgabe" #generate_text_zu_bild(file, prompt, k, rag_option, chatbot)
+            else:
+                result = "Ausgabe File" #generate_text_zu_doc(file, prompt, k, rag_option, chatbot)
+                
             history = history + [((file,), None),(prompt, result)]
 
         chatbot[-1][1] = ""
@@ -416,7 +426,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                 with gr.Row():
                         #file_display = gr.File(visible=False)
                         image_display = gr.Image( visible=False)
-                        upload = gr.UploadButton("📁", file_types=["image", "pdf"], scale = 10)
+                        upload = gr.UploadButton("📁", file_types=["image", "pdf", "docx", "pptx", "xlsx"], scale = 10)
                         emptyBtn = gr.ClearButton([user_input, chatbot, history, attached_file, image_display], value="🧹 Neue Session", scale=10)  
     
             with gr.Column():