Update app.py

app.py

@@ -43,16 +43,25 @@ def generate_image(prompt, max_retries=5):
     # This line should never be reached, but just in case:
     return None, "Unexpected error occurred"
 
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=[
-        gr.Image(label="Generated Image"),
-        gr.Textbox(label="Output Information", lines=5)
-    ],
-    title="Text-to-Image Generation",
-    description="Enter a prompt to generate an image. The system will automatically retry up to 5 times if an error occurs."
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# Text-to-Image Generation")
+    gr.Markdown("Enter a prompt to generate an image. The system will automatically retry up to 5 times if an error occurs.")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            input_text = gr.Textbox(label="Enter your prompt")
+            generate_btn = gr.Button("Generate")
+            image_output = gr.Image(label="Generated Image")
+            info_output = gr.Textbox(label="Output Information", lines=5)
+        
+        with gr.Column(scale=1):
+            file_explorer = gr.FileExplorer(label="File System")
+    
+    generate_btn.click(
+        generate_image,
+        inputs=input_text,
+        outputs=[image_output, info_output]
+    )
 
 logger.info("Launching the Gradio interface...")
-iface.launch()
+demo.launch()