Update app.py

app.py

@@ -2,6 +2,7 @@
 
 import os
 import json
+import uuid
 import gradio as gr
 
 from google import genai
@@ -22,6 +23,9 @@ SYSTEM_INSTRUCTION = (
     "Do not include extra text or markdown."
 )
 
+# In-memory session store: maps session IDs to state dicts
+state_store = {}
+
 
 def start_app(gemini_key, hf_token, hf_username, repo_name):
     os.makedirs(WORKSPACE_DIR, exist_ok=True)
@@ -102,18 +106,17 @@ def handle_message(user_msg, state):
 
 # ——— Gradio UI ———
 with gr.Blocks() as demo:
-    # Two-column layout: sidebar and main area
     with gr.Row():
         with gr.Column(scale=1):
             gemini_key = gr.Textbox(label="Gemini API Key", type="password")
             hf_token   = gr.Textbox(label="Hugging Face Token", type="password")
             hf_user    = gr.Textbox(label="HF Username")
             repo_name  = gr.Textbox(label="New App (repo) name")
+            session_id = gr.Textbox(value="", visible=False)
             start_btn  = gr.Button("Start a new app")
+
         with gr.Column(scale=3):
-            # Specify Chatbot type to avoid schema errors
             chatbot = gr.Chatbot(type="messages")
-            state   = gr.State(value=None)
             logs_display   = gr.Textbox(label="Operation Logs", interactive=False, lines=8)
             preview_iframe = gr.HTML("<p>No deployed app yet.</p>")
 
@@ -121,36 +124,41 @@ with gr.Blocks() as demo:
             send_btn = gr.Button("Send")
 
     def on_start(g_key, h_token, h_user, r_name):
-        s = start_app(g_key, h_token, h_user, r_name)
-        logs = "\n".join(s["logs"])
-        return s, logs, "<p>Awaiting first instruction...</p>"
+        new_id = str(uuid.uuid4())
+        state = start_app(g_key, h_token, h_user, r_name)
+        state_store[new_id] = state
+        logs = "\n".join(state["logs"])
+        return new_id, logs, "<p>Awaiting first instruction...</p>"
 
     start_btn.click(
         on_start,
         inputs=[gemini_key, hf_token, hf_user, repo_name],
-        outputs=[state, logs_display, preview_iframe]
+        outputs=[session_id, logs_display, preview_iframe]
     )
 
-    def on_send(msg, chat_history, s):
-        if s is None:
-            return chat_history, s, "", ""
-        reply, new_state = handle_message(msg, s)
+    def on_send(msg, chat_history, sess_id):
+        if not sess_id or sess_id not in state_store:
+            err = "Error: No API found. Please start a new app."
+            return chat_history + [("", err)], sess_id, "", ""
+        state = state_store[sess_id]
+        reply, new_state = handle_message(msg, state)
+        state_store[sess_id] = new_state
         chat_history = chat_history + [(msg, reply)]
         logs = "\n".join(new_state.get("logs", []))
         embed = ""
         if new_state.get("embed_url"):
             embed = f'<iframe src="{new_state["embed_url"]}" width="100%" height="500px"></iframe>'
-        return chat_history, new_state, logs, embed
+        return chat_history, sess_id, logs, embed
 
     send_btn.click(
         on_send,
-        inputs=[user_msg, chatbot, state],
-        outputs=[chatbot, state, logs_display, preview_iframe]
+        inputs=[user_msg, chatbot, session_id],
+        outputs=[chatbot, session_id, logs_display, preview_iframe]
     )
     user_msg.submit(
         on_send,
-        inputs=[user_msg, chatbot, state],
-        outputs=[chatbot, state, logs_display, preview_iframe]
+        inputs=[user_msg, chatbot, session_id],
+        outputs=[chatbot, session_id, logs_display, preview_iframe]
     )
 
     demo.launch()