Update app.py

app.py

@@ -59,7 +59,7 @@ def translate_text(input_text, src_lang="en"):
     return translated_text
 
 # Function to generate text using the GPT-Neo model
-def generate_text(prompt, max_length=150):
+def generate_text(prompt, max_length=50):
     generated_texts = text_generator(prompt, max_length=max_length, num_return_sequences=1)
     return generated_texts[0]["generated_text"]
 
@@ -70,10 +70,23 @@ def app_interface(input_text, src_language="en"):
     generated_text = generate_text(translated_text)
     return generated_text, generated_image
 
-# Launch the Gradio App
-gr.Interface(
-    fn=app_interface,
-    inputs=[gr.inputs.Textbox(lines=2, placeholder="Enter text here..."), gr.inputs.Dropdown(["en", "fr", "de", "es"], default="en", label="Source Language")],
-    outputs=[gr.outputs.Textbox(label="Generated Text"), gr.outputs.Image(label="Generated Image")],
-    title="Multilingual Text-to-Image & Text Generation"
-).launch()
+# Launch the Gradio App using the new Gradio components
+with gr.Blocks() as demo:
+    gr.Markdown("# Multilingual Text-to-Image & Text Generation")
+
+    # Define Gradio components
+    input_text = gr.Textbox(lines=2, placeholder="Enter text here...")
+    src_language = gr.Dropdown(["en", "fr", "de", "es"], value="en", label="Source Language")
+    
+    # Display outputs for text and image generation
+    generated_text_output = gr.Textbox(label="Generated Text")
+    generated_image_output = gr.Image(label="Generated Image")
+
+    # Button to trigger the processing
+    generate_button = gr.Button("Generate")
+
+    # Link the button to the function call
+    generate_button.click(fn=app_interface, inputs=[input_text, src_language], outputs=[generated_text_output, generated_image_output])
+
+# Run the app
+demo.launch()