Update app.py

app.py

@@ -1,60 +1,38 @@
-
 import openai
-import gradio as gr
-from gradio.components import Textbox, Image, Label
+import gradio
+
+# Initialize OpenAI API credentials
+openai.api_key = "YOUR_API_KEY"
 
-# Set up OpenAI API credentials
-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)
 
-# Define the OpenAI prompt completion function
+# Define a function to generate text response using OpenAI API
 def generate_text(prompt):
     response = openai.Completion.create(
-        model= "text-davinci-003",
+        engine="davinci",
         prompt=prompt,
-        max_tokens=60,
+        max_tokens=1024,
         n=1,
         stop=None,
-        temperature=0.8,
+        temperature=0.7,
     )
     return response.choices[0].text.strip()
 
-# Define the image generation function
-model_path = "models/dreamlike-art/dreamlike-photoreal-2.0"
-model_prefix = "photoreal style"
-model = gr.Interface.load(model_path)
-
-def generate_image(prompt):
-    # Generate text using OpenAI's GPT-3 model
-    text = generate_text(prompt)
-
-    # Generate an image using a DALL-E model trained on photorealistic images
-    if model is not None:
-        image_return = model.predict(model_prefix + ": " + text)
-    else:
-        image_return = None
-
-    # Return the generated image
-    return image_return[0] if image_return is not None else None
-
-# Define the Gradio interface
-prompt_input = Textbox("Enter a prompt", lines=3)
-text_output = Label("")
-image_output = Image()
-
-iface = gr.Interface(
-    generate_image,
-    inputs=prompt_input,
-    outputs=[text_output, image_output],
-    title="Image Generator",
-    description="Generates an image based on a prompt using OpenAI's GPT-3 and DALL-E models."
+# Define a function to generate image from text prompt using the loaded model
+def process_prompt(prompt):
+    prompt += ", in " + model_prefix
+    image_return = model(prompt)
+    return image_return
+
+# Create a Gradio interface with an input textbox for the prompt and output sections for image and text
+interface = gradio.Interface(
+    fn=lambda prompt: (process_prompt(prompt), generate_text(prompt)),
+    inputs=gradio.inputs.Textbox(label="Enter Prompt"),
+    outputs=[gradio.outputs.Image(label="Generated Image"), gradio.outputs.Textbox(label="Generated Text")]
 )
 
-# Update the output boxes with the generated text and image
-def update_text_output(text, image_url):
-    text_output.text = text
-    text_output.style['font-size'] = '8px'  # Change the font size here
-
-iface.interface_result_hooks = [update_text_output]
-
 # Launch the interface
-iface.launch()
+interface.launch()