Update app.py

app.py

@@ -43,25 +43,16 @@ def generate_image(prompt, max_retries=5):
     # This line should never be reached, but just in case:
     return None, "Unexpected error occurred"
 
-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]
-    )
+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."
+)
 
 logger.info("Launching the Gradio interface...")
-demo.launch()
+iface.launch()