Update app.py

app.py

@@ -22,21 +22,30 @@ def generate_text(prompt):
     return response.choices[0].text.strip()
 
 # Define a function to generate image from text prompt using the loaded model
-def process_prompt(prompt):
-    prompt += ", in " + model_prefix
+def process_prompt(text):
+    prompt = text + ", in " + model_prefix
     image_return = model(prompt)
     return image_return
 
+# Create Gradio interface
 interface = gradio.Interface(
     fn=process_prompt,
     inputs=["text", gradio.inputs.Textbox(label="Enter prompt")],
     outputs=[
-        "image", gradio.outputs.Image(label="Generated Image", type="pil"),
+        "image", gradio.outputs.Image(label="Generated Image", type="numpy"),
         "text", gradio.outputs.Textbox(label="Generated Text")
     ],
     title="AlStable Text-to-Image",
     description="Generates an image from a text prompt and provides text response.",
 )
 
+# Add a callback function to update the "Generated Text" output box based on the input text
+def update_output_text(inp):
+    response_text = generate_text(inp)
+    interface.set_output_text(response_text)
+
+# Set the callback function for the input textbox
+interface.inputs[0].set_property("on_input", update_output_text)
+
 # Launch the interface
 interface.launch()