Update app_2.py

app_2.py

@@ -1,9 +1,10 @@
-import os, mimetypes, threading
+import os
+import threading
 import gradio as gr
 from google import genai
 from google.genai import types
 
-client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
 MODEL = "gemini-3.1-flash-lite-preview"
 
 # Tools + Config (einmalig)
@@ -14,39 +15,39 @@ generate_content_config = types.GenerateContentConfig(
 )
 
 def respond(message, history, state):
-    # history: [{"role": "user"/"assistant", "content": "..."} , ...]
     if state is None:
         state = {
             "chat": client.chats.create(model=MODEL, config=generate_content_config),
             "lock": threading.Lock(),
         }
-
+    
     history = history or []
-    text = (message or {}).get("text") or ""
-    files = (message or {}).get("files") or []
-
+    text = (message or {}).get("text", "")
+    files = (message or {}).get("files", [])
     parts = []
-    if text:
-        parts.append(types.Part.from_text(text=text))
 
+    # 1. Dateien verarbeiten (Upload via API & UI-Vorbereitung)
     for file_path in files:
-        mime, _ = mimetypes.guess_type(file_path)
-        mime = mime or "application/octet-stream"
-        with open(file_path, "rb") as f:
-            parts.append(types.Part.from_bytes(data=f.read(), mime_type=mime))
+        # Datei an Gemini Server senden
+        uploaded_file = client.files.upload(file=file_path)
+        parts.append(uploaded_file)
+        # Datei in der Gradio UI als Tupel anzeigen
+        history.append({"role": "user", "content": (file_path,)})
+
+    # 2. Text verarbeiten
+    if text:
+        parts.append(text)
+        history.append({"role": "user", "content": text})
 
     if not parts:
         yield history, state
         return
 
-    # UI: user msg hinzufügen
-    user_label = text if text else "[Datei]"
-    history.append({"role": "user", "content": user_label})
-    # UI: assistant placeholder
+    # 3. UI: Assistant Placeholder hinzufügen
     history.append({"role": "assistant", "content": ""})
     yield history, state
 
-    # Gemini: send + stream (Historie steckt im state["chat"])
+    # 4. Gemini: Senden + Streamen
     with state["lock"]:
         out = ""
         for chunk in state["chat"].send_message_stream(parts):
@@ -59,18 +60,24 @@ def clear_msg():
     return {"text": "", "files": []}
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Gemini Thinking AI")
+    gr.Markdown("# Gemini AI Chat (Gradio 6.x & File API)")
+    
+    # Standard-Initialisierung für Gradio >= 6.0
     chatbot = gr.Chatbot(height=600)
+    
     msg = gr.MultimodalTextbox(
         placeholder="Schreib was oder lade Dateien hoch…",
         file_count="multiple",
     )
     state = gr.State(None)
-
+    
     msg.submit(respond, inputs=[msg, chatbot, state], outputs=[chatbot, state]) \
        .then(clear_msg, outputs=msg)
+       
+    gr.Button("Clear").click(
+        lambda: ([], None, {"text": "", "files": []}),
+        outputs=[chatbot, state, msg]
+    )
 
-    gr.Button("Clear").click(lambda: ([], None, {"text": "", "files": []}),
-                             outputs=[chatbot, state, msg])
-
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()