Update app.py

app.py

@@ -4,40 +4,55 @@ from google import genai
 from google.genai import types
 
 # Client initialisieren
-client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
+client_genai = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
 
-def generate(prompt, history):
-    # Vorbereitung der Inhalte
+def generate_for_js(input_text):
+    """
+    Diese Funktion entspricht exakt der Erwartung des JS-Clients:
+    - Input: input_text
+    - Output: Ein String (oder Liste von Strings)
+    """
     contents = [
         types.Content(
             role="user",
-            parts=[types.Part.from_text(text=prompt)],
+            parts=[types.Part.from_text(text=input_text)],
         ),
     ]
     
-    # Konfiguration mit Google Search Tool
     tools = [types.Tool(google_search=types.GoogleSearch())]
     config = types.GenerateContentConfig(tools=tools)
 
-    response_text = ""
+    full_response = ""
     
-    # Streaming-Generator für Gradio
-    for chunk in client.models.generate_content_stream(
-        model="gemini-2.0-flash-001", # Aktuellste Version
-        contents=contents,
-        config=config,
-    ):
-        if chunk.text:
-            response_text += chunk.text
-            yield response_text
+    try:
+        # Da das JS await client.predict nutzt, sammeln wir hier den gesamten Text
+        # Falls du echtes Streaming im JS willst, müsste dort 'client.submit' genutzt werden.
+        response = client_genai.models.generate_content(
+            model="gemini-2.0-flash-001",
+            contents=contents,
+            config=config,
+        )
+        return response.text
+    except Exception as e:
+        return f"Fehler: {str(e)}"
 
-# Gradio Interface Setup
-demo = gr.ChatInterface(
-    fn=generate,
-    title="Gemini Flash mit Google Search",
-    description="Gib eine Frage ein, und Gemini nutzt die Google Suche für aktuelle Antworten.",
-    type="messages"
-)
+# Gradio Interface Definition
+with gr.Blocks() as demo:
+    # Der API-Name muss "generate" sein, damit /generate im JS funktioniert
+    prompt_input = gr.Textbox(label="Input Text", visible=False)
+    output_text = gr.Markdown(visible=False)
+    
+    submit_btn = gr.Button("Senden", visible=False)
+    
+    # Hier definieren wir den Endpunkt explizit
+    submit_btn.click(
+        fn=generate_for_js, 
+        inputs=prompt_input, 
+        outputs=output_text, 
+        api_name="generate"
+    )
 
 if __name__ == "__main__":
+    # Wichtig: Wenn du es auf Hugging Face Spaces hostest (mgokg/webapi), 
+    # wird der api_name automatisch unter /predict/generate verfügbar.
     demo.launch()