Update app.py

app.py

@@ -1,7 +1,6 @@
+# This is a Gradio app that provides a text-based chat interface with optional image and video generation.
 import gradio as gr
-import torch
-from transformers import pipeline, set_seed
-from diffusers import DiffusionPipeline
+import numpy as np
 import tempfile
 import imageio
 
@@ -12,29 +11,29 @@ AVAILABLE_MODELS = {
     "Mistral (OpenAccess)": "mistralai/Mistral-7B-v0.1"
 }
 
-set_seed(42)
-text_model_cache = {}
-
 # Load text-to-image model using diffusers (correct API)
 try:
+    from diffusers import DiffusionPipeline
     image_generator = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
     image_generator.to("cpu")
     image_enabled = True
-except Exception as e:
+except ImportError:
     image_generator = None
     image_enabled = False
-    print(f"[Image model error]: {e}")
+    print("[Image model error]: diffusers library not found")
 
 # Load text-to-video model
 try:
+    from diffusers import DiffusionPipeline
     video_pipeline = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b")
     video_pipeline.to("cpu")
     video_enabled = True
-except Exception as e:
+except ImportError:
     video_pipeline = None
     video_enabled = False
-    print(f"[Video model error]: {e}")
+    print("[Video model error]: diffusers library not found")
 
+text_model_cache = {}
 chat_memory = {}
 
 # ---------- Core Function ----------
@@ -46,15 +45,18 @@ def codette_terminal(prompt, model_name, generate_image, generate_video, session
         chat_memory[session_id] = []
         return "🧠 Codette signing off... Session reset.", None, None
 
+    # Load the text generation model if it's not already in the cache
     if model_name not in text_model_cache:
-        text_model_cache[model_name] = pipeline("text-generation", model=AVAILABLE_MODELS[model_name])
+        text_model_cache[model_name] = gr.pipelines.TextGeneration(model=AVAILABLE_MODELS[model_name])
     generator = text_model_cache[model_name]
     response = generator(prompt, max_length=100, num_return_sequences=1, do_sample=True)[0]['generated_text'].strip()
 
+    # Update the chat log
     chat_memory[session_id].append(f"🖋️ You > {prompt}")
     chat_memory[session_id].append(f"🧠 Codette > {response}")
     chat_log = "\n".join(chat_memory[session_id][-10:])
 
+    # Generate image if requested and image generation is enabled
     img = None
     if generate_image and image_enabled:
         try:
@@ -62,6 +64,7 @@ def codette_terminal(prompt, model_name, generate_image, generate_video, session
         except Exception as e:
             chat_log += f"\n[Image error]: {e}"
 
+    # Generate video if requested and video generation is enabled
     vid = None
     if generate_video and video_enabled:
         try:
@@ -95,4 +98,4 @@ with gr.Blocks(title="Codette Terminal – Text + Image + Video") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(show_error=True)