Update app.py

app.py

@@ -1,51 +1,36 @@
-import openai
 import gradio
+import openai
 
-# Initialize OpenAI API credentials
-openai.api_key = openai.api_key
+openai.api_key = "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"
 
-# Load the text-to-image generation model
 model_path = "models/dreamlike-art/dreamlike-photoreal-2.0"
 model_prefix = "photoreal style"
-model = gradio.Interface.load(model_path)
+model = gradio.load_model(model_path)
 
-# Define a function to generate text response using OpenAI API
 def generate_text(prompt):
     response = openai.Completion.create(
-        engine="davinci",
-        prompt=prompt,
-        max_tokens=1024,
-        n=1,
-        stop=None,
-        temperature=0.7,
+      engine="davinci",
+      prompt=prompt,
+      max_tokens=1024,
+      n=1,
+      stop=None,
+      temperature=0.7,
     )
-    return response.choices[0].text.strip()
+    return response.choices[0].text
 
-# Define a function to generate image from text prompt using the loaded model
-def process_prompt(text):
-    prompt = text + ", in " + model_prefix
+def process_prompt(prompt):
+    prompt += ", 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="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)
+def update_output_text(prompt):
+    text = generate_text(prompt)
+    interface.outputs[0].set_text(text)
+    output_image = process_prompt(text)
+    interface.outputs[1].set_image(output_image)
 
-# Launch the interface
+interface = gradio.Interface(fn=update_output_text, 
+                        inputs=[gradio.inputs.Textbox(label="Enter Prompt:")],
+                        outputs=[gradio.outputs.Textbox(label="Generated Text"), gradio.outputs.Image(label="Generated Image")], 
+                        title='AlStable Text to Image')
 interface.launch()