Update app.py

app.py

@@ -5,6 +5,7 @@ import base64
 from io import BytesIO
 from PIL import Image
 import numpy as np
+import traceback
 
 # Initialize the Together client
 api_key = os.environ.get('TOGETHER_API_KEY')
@@ -58,13 +59,17 @@ def generate_gradio_app(image):
         for chunk in stream:
             if chunk.choices[0].delta.content is not None:
                 generated_text += chunk.choices[0].delta.content
-                yield generated_text  # Yield partial results for gradio to update in real-time
+                yield f"Generating... (Current length: {len(generated_text)} characters)\n\n{generated_text}"
+        
+        if not generated_text:
+            return "Error: No response generated from the model. Please try again."
         
         return generated_text
 
     except Exception as e:
         error_message = str(e)
-        return f"An error occurred: {error_message}\nPlease try again or check your API key and connection."
+        stack_trace = traceback.format_exc()
+        return f"An error occurred: {error_message}\n\nStack trace:\n{stack_trace}\n\nPlease try again or check your API key and connection."
 
 with gr.Blocks() as demo:
     gr.Markdown("# Analyze wireframe and suggest Gradio app layout")
@@ -84,4 +89,5 @@ with gr.Blocks() as demo:
         outputs=[text_output]
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch(debug=True)