Update app.py

app.py

@@ -1,50 +1,68 @@
 import gradio as gr
 from groq import Groq
 
-def generate_and_preview(api_key, prompt):
+# Speichert die Konversation pro Session
+chat_history = []
+
+def chat_game(api_key, user_input):
     """
-    Generiert HTML5-Spiel mit Groq und gibt Code + Vorschau zurück.
-    Vorschau wird über ein iframe dargestellt, damit JS läuft.
+    Chat für HTML5 Game Coding mit Follow-up. 
+    Gibt Code + Live Preview zurück.
     """
     if not api_key:
         return "Bitte API-Key eingeben!", "<p>Keine Vorschau</p>"
-    
+
     try:
         client = Groq(api_key=api_key)
+
+        # System Prompt definiert die Rolle der KI
+        system_prompt = {
+            "role": "system",
+            "content": (
+                "Du bist ein professioneller Game Developer und AI-Coding-Assistent. "
+                "Erzeuge oder korrigiere HTML5-Spiele. "
+                "Nur gültiger HTML+JS Code, keine Erklärungen. "
+                "Canvas-basiertes Spiel, sofort spielbar."
+            )
+        }
+
+        # User Prompt hinzufügen
+        user_message = {"role": "user", "content": user_input}
+
+        # Alle vorherigen Messages + neue Nachricht
+        messages = [system_prompt] + chat_history + [user_message]
+
+        # Anfrage an Groq
         completion = client.chat.completions.create(
-            model="openai/gpt-oss-120b",
-            messages=[
-                {
-                    "role": "system",
-                    "content": (
-                        "Du bist ein professioneller Game Developer. "
-                        "Generiere vollständigen HTML5-Spielcode. "
-                        "Nur HTML + JS, kein Markdown oder Text. "
-                        "Canvas-basiertes Spiel, sofort spielbar."
-                    )
-                },
-                {"role": "user", "content": prompt}
-            ],
+            model="llama3-70b-8192",  # oder gpt-os wenn verfügbar
+            messages=messages,
             temperature=0.8
         )
-        html_code = completion.choices[0].message.content
 
-        # Vorschau in iframe packen, damit JS läuft
+        # Antwort des Models
+        model_response = completion.choices[0].message.content
+
+        # Antwort zur History hinzufügen
+        chat_history.append(user_message)
+        chat_history.append({"role": "assistant", "content": model_response})
+
+        # Vorschau im iframe
         preview_html = f"""
-        <iframe srcdoc="{html_code.replace('"', '&quot;')}" 
+        <iframe srcdoc="{model_response.replace('"', '&quot;')}" 
                 width="100%" height="600px" style="border:none;"></iframe>
         """
-        return html_code, preview_html
+
+        return model_response, preview_html
 
     except Exception as e:
         return f"Fehler: {e}", "<p>Fehler bei der Vorschau</p>"
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# ⚡ KI HTML5 Game Generator mit Live-Vorschau")
+    gr.Markdown("# 🤖 HTML5 Game AI Chat mit Live Preview")
     gr.Markdown(
-        "API-Key eingeben (nicht gespeichert) + Spielidee → "
-        "Die KI generiert HTML5-Code. Vorschau rechts neben dem Code."
+        "API-Key eingeben (nicht gespeichert) und eine Spielidee oder Folgefrage stellen. "
+        "Die KI generiert HTML5-Code und zeigt Live-Vorschau rechts."
     )
 
     with gr.Row():
@@ -54,12 +72,12 @@ with gr.Blocks() as demo:
                 placeholder="Hier API-Key einfügen",
                 type="password"
             )
-            prompt_input = gr.Textbox(
-                label="Spielidee",
+            user_input = gr.Textbox(
+                label="Deine Frage / Spielidee",
                 placeholder="Beispiel: 2D Jump'n'Run mit Gegnern",
                 lines=3
             )
-            btn = gr.Button("🎮 Spiel generieren")
+            btn = gr.Button("🎮 Senden")
             code_output = gr.Code(
                 language="html",
                 label="HTML5 Game Code",
@@ -69,9 +87,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             preview_output = gr.HTML(label="Live Vorschau")
 
-    # Event: Button klick
-    btn.click(generate_and_preview, 
-              inputs=[api_key_input, prompt_input], 
-              outputs=[code_output, preview_output])
+    # Event
+    btn.click(chat_game, inputs=[api_key_input, user_input], outputs=[code_output, preview_output])
 
 demo.launch()