Update app.py

app.py

@@ -3,56 +3,72 @@ import gradio as gr
 from google import genai
 from google.genai import types
 
-# Client initialisieren
+# Client initialization
 client_genai = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
 
-def generate_for_js(input_text):
+def generate_with_history(input_text, history):
     """
-    Diese Funktion entspricht exakt der Erwartung des JS-Clients:
-    - Input: input_text
-    - Output: Ein String (oder Liste von Strings)
+    - input_text: The new message from JS
+    - history: A list of previous messages [user, model, user, model...]
     """
-    contents = [
-        types.Content(
-            role="user",
-            parts=[types.Part.from_text(text=input_text)],
-        ),
-    ]
+    if history is None:
+        history = []
+
+    # Build the contents list including history
+    # Gemini API expects a list of Content objects
+    contents = []
+    for msg in history:
+        contents.append(types.Content(
+            role=msg["role"],
+            parts=[types.Part.from_text(text=msg["content"])]
+        ))
     
+    # Add the current user prompt
+    contents.append(types.Content(
+        role="user",
+        parts=[types.Part.from_text(text=input_text)]
+    ))
+
     tools = [types.Tool(google_search=types.GoogleSearch())]
     config = types.GenerateContentConfig(tools=tools)
 
-    full_response = ""
-    
     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
+        
+        response_text = response.text
+        
+        # Update history: add the current exchange
+        history.append({"role": "user", "content": input_text})
+        history.append({"role": "model", "content": response_text})
+        
+        # Return the response for JS and the updated history for the State
+        return response_text, history
+
     except Exception as e:
-        return f"Fehler: {str(e)}"
+        return f"Error: {str(e)}", history
 
-# Gradio Interface Definition
+# Building the Blocks interface
 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)
+    # State component to store the conversation across API calls
+    history_state = gr.State([])
     
-    submit_btn = gr.Button("Senden", visible=False)
+    # Hidden components to match your JS expectations
+    input_box = gr.Textbox(visible=False)
+    output_box = gr.Markdown(visible=False)
     
-    # Hier definieren wir den Endpunkt explizit
-    submit_btn.click(
-        fn=generate_for_js, 
-        inputs=prompt_input, 
-        outputs=output_text, 
+    # The API endpoint
+    # Note: We include history_state in both inputs and outputs
+    btn = gr.Button("Submit", visible=False)
+    btn.click(
+        fn=generate_with_history,
+        inputs=[input_box, history_state],
+        outputs=[output_box, history_state],
         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()