Update app.py

app.py

@@ -73,6 +73,7 @@ def generate_motion(text, vqvae_model, transformer_model):
     return pred_xyz.cpu().numpy().reshape(-1, 22, 3)
 
 def infer(text):
+    print("Received text:", text)  # Debug: Confirming that the text input is correctly received
     try:
         motion_data = generate_motion(text, vqvae_model, transformer_model)
         if motion_data.size == 0:
@@ -84,11 +85,13 @@ def infer(text):
     try:
         gif_data = draw_to_batch([motion_data], [text], None)
         if gif_data:
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".gif") as temp_file:
-                temp_file.write(gif_data)
-                temp_file_path = temp_file.name
-            print("GIF successfully saved to temporary file.")
-            return temp_file_path
+            # Save the GIF to a permanent file rather than a temporary file for debugging
+            gif_filename = "output.gif"
+            gif_path = os.path.join(tempfile.gettempdir(), gif_filename)
+            with open(gif_path, "wb") as gif_file:
+                gif_file.write(gif_data)
+            print("GIF successfully saved to:", gif_path)  # Debug: Check the exact path of the saved GIF
+            return gif_path
         else:
             print("Failed to generate GIF data.")
             return "Error generating GIF. Please try again."
@@ -96,11 +99,12 @@ def infer(text):
         print(f"Error generating GIF: {str(e)}")
         return "Error generating GIF. Please try again."
 
-with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
+css = ".container { max-width: 800px; margin: auto; }"
+with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("## 3D Human Motion Generation")
         with gr.Row():
-            text_input = gr.Textbox(label="Enter the human action to generate", placeholder="Enter text description for the action here...", show_label=False, max_lines=1, container=False)
+            text_input = gr.Textbox(label="Enter the human action to generate", placeholder="Enter text description for the action here...", show_label=True)
             submit_button = gr.Button("Generate Motion")
         output_image = gr.Image(label="Generated Human Motion", type="filepath", show_label=False)
 
@@ -109,6 +113,6 @@ with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
         inputs=[text_input],
         outputs=[output_image]
     )
-
+    
 if __name__ == "__main__":
     demo.launch()