Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import tempfile
 import imageio
 import torch
+import time
 from transformers import pipeline
 from diffusers import DiffusionPipeline
 
@@ -75,7 +76,7 @@ def codette_terminal(prompt, model_name, generate_image, generate_video, session
         yield f"[Text generation error]: {e}", None, None
         return
 
-    # Stream the output character by character
+    # Stream the output
     response_so_far = ""
     for char in output:
         response_so_far += char
@@ -83,14 +84,12 @@ def codette_terminal(prompt, model_name, generate_image, generate_video, session
         temp_log.append(f"🖋️ You > {prompt}")
         temp_log.append(f"🧠 Codette > {response_so_far}")
         yield "\n".join(temp_log[-10:]), None, None
-        import time
         time.sleep(0.01)
 
     # Finalize chat memory
     chat_memory[session_id].append(f"🖋️ You > {prompt}")
     chat_memory[session_id].append(f"🧠 Codette > {output}")
 
-    # Image Generation
     imgs = None
     if generate_image and image_enabled:
         try:
@@ -99,7 +98,6 @@ def codette_terminal(prompt, model_name, generate_image, generate_video, session
         except Exception as e:
             response_so_far += f"\n[Image error]: {e}"
 
-    # Video Generation
     vid = None
     if generate_video and video_enabled:
         try:
@@ -118,25 +116,31 @@ with gr.Blocks(title="🧬 Codette Terminal – Streamed AI Chat") as demo:
     gr.Markdown("## 🧬 Codette Terminal (Chat + Image + Video + Batch + NSFW OK)")
     gr.Markdown("Type a prompt, select your model, and configure generation options. Type `'exit'` to reset.")
 
-    session_id = gr.Textbox(value="session_default", visible=False)
-    model_dropdown = gr.Dropdown(choices=list(AVAILABLE_MODELS.keys()), value="GPT-2 (small, fast)", label="Language Model")
-    generate_image_toggle = gr.Checkbox(label="Generate Image(s)?", value=False, interactive=image_enabled)
-    generate_video_toggle = gr.Checkbox(label="Generate Video?", value=False, interactive=video_enabled)
-    batch_size_slider = gr.Slider(label="Number of Images", minimum=1, maximum=4, step=1, value=1)
-    video_steps_slider = gr.Slider(label="Video Inference Steps", minimum=10, maximum=100, step=10, value=50)
-    fps_slider = gr.Slider(label="Video FPS", minimum=4, maximum=24, step=2, value=8)
+    with gr.Row():
+        session_id = gr.Textbox(value="session_default", visible=False)
+        model_dropdown = gr.Dropdown(choices=list(AVAILABLE_MODELS.keys()), value="GPT-2 (small, fast)", label="Language Model")
+
+    with gr.Row():
+        generate_image_toggle = gr.Checkbox(label="Generate Image(s)?", value=False, interactive=image_enabled)
+        generate_video_toggle = gr.Checkbox(label="Generate Video?", value=False, interactive=video_enabled)
+
+    with gr.Row():
+        batch_size_slider = gr.Slider(label="Number of Images", minimum=1, maximum=4, step=1, value=1)
+        video_steps_slider = gr.Slider(label="Video Inference Steps", minimum=10, maximum=100, step=10, value=50)
+        fps_slider = gr.Slider(label="Video FPS", minimum=4, maximum=24, step=2, value=8)
+
     user_input = gr.Textbox(label="Your Prompt", placeholder="e.g. A robot dreaming on Mars", lines=1)
     output_text = gr.Textbox(label="Codette Output", lines=15, interactive=False)
-    output_image = gr.Gallery(label="Generated Image(s)").style(grid=2)
+    output_image = gr.Gallery(label="Generated Image(s)", columns=2)
     output_video = gr.Video(label="Generated Video")
 
     user_input.submit(
         codette_terminal,
-        inputs=[user_input, model_dropdown, generate_image_toggle, generate_video_toggle, session_id, batch_size_slider, video_steps_slider, fps_slider],
-        outputs=[output_text, output_image, output_video],
-        concurrency_limit=1,
-        queue=True,
-        show_progress=True
+        inputs=[
+            user_input, model_dropdown, generate_image_toggle, generate_video_toggle,
+            session_id, batch_size_slider, video_steps_slider, fps_slider
+        ],
+        outputs=[output_text, output_image, output_video]
     )
 
 # ---------- Launch ----------