fixed again app.py

app.py

@@ -9,11 +9,9 @@ from gradio_pannellum import Pannellum
 import io
 from huggingface_hub import snapshot_download
 from txt2panoimg import Text2360PanoramaImagePipeline
-from transformers import utils
 
 
 # Download the model
-utils.move_cache()
 model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
 
 # Initialize pipelines
@@ -24,11 +22,12 @@ def text_to_pano(prompt, upscale):
     try:
         input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
         output = txt2panoimg(input_data)
-        return output, output 
+        return output, output
     except Exception as e:
         print(f"Error generating panorama: {e}")
         return None, None
 
+
 title = """
 <div style="text-align: center;">
     <img src="https://ant.dpu.ac.th/wp-content/uploads/2024/04/dpulogo.png" width="200"/>
@@ -60,9 +59,11 @@ def create_aframe_html(image):
     """
     return aframe_html
 
+
 def create_aframe_preview(image):
-    if image is None:  # Handle case where image generation failed
+    if image is None: 
         return gr.HTML("Error generating image. Please check the prompt or try again.")
+
     buffered = io.BytesIO()
     image.save(buffered, format="JPEG")
     img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
@@ -75,9 +76,10 @@ def create_aframe_preview(image):
     """
     return gr.HTML(f'<iframe srcdoc="{aframe_html}" allowfullscreen frameborder="0"></iframe>')
 
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.HTML(title)
-    
+
     with gr.Row():
         with gr.Column():
             t2p_input = gr.Textbox(label="Enter your prompt", lines=3)
@@ -96,20 +98,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     def generate_with_update(prompt, upscale, trigger):
         output, image = text_to_pano(prompt, upscale)
-        if image is not None:  # Only create HTML if image generation succeeded
-            html = create_aframe_html(image)
-        else:
-            html = "Failed to generate A-Frame HTML. Please try again."
-        return output, image, html, create_aframe_preview(image), trigger + 1
+        html = create_aframe_html(image) if image is not None else "Failed to generate A-Frame HTML. Please try again."
+        preview = create_aframe_preview(image)
+        return output, image, html, preview, trigger + 1
 
     t2p_generate.click(
         generate_with_update,
         inputs=[t2p_input, t2p_upscale, update_trigger],
         outputs=[t2p_output, t2p_image_output, output_html, aframe_preview, update_trigger]
     )
-    output_image.change(create_aframe_preview, inputs=[output_image], outputs=[aframe_preview])
-
-
+    t2p_image_output.change(create_aframe_preview, inputs=[t2p_image_output], outputs=[aframe_preview])
+    
     copy_html_button.click(
         None,
         inputs=[output_html],