Update app.py

app.py

@@ -5,6 +5,7 @@ import json
 import datetime
 import copy
 
+# Ordner für lokale Projekte
 PROJECT_DIR = "projects"
 os.makedirs(PROJECT_DIR, exist_ok=True)
 
@@ -29,6 +30,9 @@ def create_checkpoint():
     }
     current_project["checkpoints"].append(checkpoint)
 
+def update_checkpoint_list():
+    return [c["timestamp"] for c in current_project["checkpoints"]] if current_project["checkpoints"] else []
+
 def new_project():
     global current_project
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -51,7 +55,7 @@ loop();
         "checkpoints": []
     }
     create_checkpoint()
-    return current_project["code"], "<p>Live Vorschau</p>", list_projects(), "Neues Projekt erstellt: " + current_project["name"]
+    return current_project["code"], "<p>Live Vorschau</p>", list_projects(), "Neues Projekt erstellt: " + current_project["name"], update_checkpoint_list()
 
 def load_project(project_name):
     global current_project
@@ -59,8 +63,11 @@ def load_project(project_name):
     if os.path.exists(filepath):
         with open(filepath,"r",encoding="utf-8") as f:
             current_project = json.load(f)
-        return current_project["code"], f"<iframe srcdoc='{current_project['code'].replace('\"','&quot;')}' width='100%' height='400px' style='border:none;'></iframe>", list_projects(), f"Projekt geladen: {project_name}", update_checkpoint_list()
-    return "", "<p>Keine Vorschau</p>", list_projects(), "Projekt nicht gefunden", []
+        html_escaped = current_project["code"].replace('"', "&quot;")
+        iframe_html = f"<iframe srcdoc='{html_escaped}' width='100%' height='400px' style='border:none;'></iframe>"
+        return current_project["code"], iframe_html, list_projects(), f"Projekt geladen: {project_name}", update_checkpoint_list()
+    else:
+        return "", "<p>Keine Vorschau</p>", list_projects(), "Projekt nicht gefunden", []
 
 def save_project():
     global current_project
@@ -104,21 +111,21 @@ def chat_generate(api_key, model_name, user_input, current_code):
         current_project["chat_history"].append({"role":"assistant","content":model_response})
         create_checkpoint()
 
-        preview_html = f"<iframe srcdoc='{model_response.replace('\"','&quot;')}' width='100%' height='400px' style='border:none;'></iframe>"
+        html_escaped = model_response.replace('"', "&quot;")
+        preview_html = f"<iframe srcdoc='{html_escaped}' width='100%' height='400px' style='border:none;'></iframe>"
 
         return model_response, preview_html, "✅ KI Update fertig", update_checkpoint_list()
     except Exception as e:
         return current_code, "<p>Fehler bei der Vorschau</p>", f"Fehler: {e}", update_checkpoint_list()
 
-def update_checkpoint_list():
-    return [f"{c['timestamp']}" for c in current_project["checkpoints"]] if current_project["checkpoints"] else []
-
 def load_checkpoint(checkpoint_time):
     for c in current_project["checkpoints"]:
-        if c["timestamp"]==checkpoint_time:
-            current_project["code"]=c["code"]
-            current_project["chat_history"]=c["chat_history"]
-            return current_project["code"], f"<iframe srcdoc='{c['code'].replace('\"','&quot;')}' width='100%' height='400px' style='border:none;'></iframe>", "✅ Checkpoint geladen"
+        if c["timestamp"] == checkpoint_time:
+            current_project["code"] = c["code"]
+            current_project["chat_history"] = c["chat_history"]
+            html_escaped = c["code"].replace('"', "&quot;")
+            iframe_html = f"<iframe srcdoc='{html_escaped}' width='100%' height='400px' style='border:none;'></iframe>"
+            return current_project["code"], iframe_html, "✅ Checkpoint geladen"
     return current_project["code"], "<p>Keine Vorschau</p>", "Checkpoint nicht gefunden"
 
 def delete_checkpoint(checkpoint_time):
@@ -136,8 +143,9 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             gr.Markdown("### KI Beschreibung / Hinweise")
             ki_desc = gr.HTML("<p>Hier erscheinen Hinweise der KI und aktuelle Anfrage</p>")
+            
             api_key_input = gr.Textbox(label="GROQ API-Key", placeholder="Hier API-Key einfügen", type="password")
-            model_dropdown = gr.Dropdown(choices=["llama3-70b-8192","gpt-os"], value="llama3-70b-8192", label="AI Modell auswählen")
+            model_input = gr.Textbox(label="AI Modell", placeholder="Beispiel: llama3-70b-8192 oder gpt-os", value="llama3-70b-8192")
             user_input = gr.Textbox(label="Spielidee / Follow-up", placeholder="Beispiel: Gegner hinzufügen", lines=3)
             
             with gr.Row():
@@ -164,11 +172,11 @@ with gr.Blocks() as demo:
     # ------------------------------
     # Events
     # ------------------------------
-    btn_new.click(new_project, outputs=[code_editor, preview_output, project_dropdown, status_output])
+    btn_new.click(new_project, outputs=[code_editor, preview_output, project_dropdown, status_output, checkpoint_dropdown])
     btn_load.click(load_project, inputs=[project_dropdown], outputs=[code_editor, preview_output, project_dropdown, status_output, checkpoint_dropdown])
     btn_save_code.click(save_code_local, inputs=[code_editor], outputs=[status_output, checkpoint_dropdown])
     btn_save_project.click(save_project, outputs=[status_output])
-    btn_ai.click(chat_generate, inputs=[api_key_input, model_dropdown, user_input, code_editor], outputs=[code_editor, preview_output, status_output, checkpoint_dropdown])
+    btn_ai.click(chat_generate, inputs=[api_key_input, model_input, user_input, code_editor], outputs=[code_editor, preview_output, status_output, checkpoint_dropdown])
     btn_load_checkpoint.click(load_checkpoint, inputs=[checkpoint_dropdown], outputs=[code_editor, preview_output, status_output])
     btn_delete_checkpoint.click(delete_checkpoint, inputs=[checkpoint_dropdown], outputs=[checkpoint_dropdown, status_output])