Update app.py

app.py

@@ -4,7 +4,6 @@ from gradio_client import Client
 from PIL import Image
 import requests
 from io import BytesIO
-import streamlit as st
 
 # Initialize the HuggingFace Inference Client with the specified model
 client_mistral = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
@@ -41,16 +40,16 @@ def generate_improved_prompt(logo_request, temperature=0.9, max_new_tokens=512,
         output += response.token.text
         yield output
 
-def generate_image(prompt, negative_prompt="", use_negative_prompt=False, seed=0, width=1024, height=1024, guidance_scale=7.5, randomize_seed=True):
+def generate_image(prompt):
     result = client_playground.predict(
         prompt,
-        negative_prompt,
-        use_negative_prompt,
-        seed,
-        width,
-        height,
-        guidance_scale,
-        randomize_seed,
+        "",  # negative prompt
+        False,  # use negative prompt
+        0,  # seed
+        1024,  # width
+        1024,  # height
+        7.5,  # guidance scale
+        True,  # randomize seed
         api_name="/run"
     )
 
@@ -75,7 +74,12 @@ css = """
 }
 """
 
-with gr.Blocks(css=css) as gpt:
+def process_request(logo_request):
+    improved_prompt = next(generate_improved_prompt(logo_request))
+    image = generate_image(improved_prompt)
+    return image
+
+with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column(scale=2):
             gr.HTML("<h1>Settings</h1>")
@@ -83,17 +87,17 @@ with gr.Blocks(css=css) as gpt:
         with gr.Column(scale=3):
             gr.HTML("<h1><center>Logo Prompt Generator<h1><center>")
             generate_button = gr.Button("Generate")
-            output_area = gr.Textbox(label="AI Response", interactive=False, lines=10)
+            output_image = gr.Image(label="Generated Image")
             generate_button.click(
-                fn=generate_improved_prompt,
+                fn=process_request,
                 inputs=[logo_input],
-                outputs=output_area
+                outputs=output_image
             )
 
     gr.Markdown("""
     ---
     ### Meta Information
-    **Project Title**: Magic AI Website Logo Creator
+    **Project Title**: Logo Prompt Generator
     
     **Github**: [https://github.com/pacnimo/gpt-prompt-generator](https://github.com/pacnimo/)
     
@@ -102,24 +106,4 @@ with gr.Blocks(css=css) as gpt:
     **Footer**: © 2024 by [pacnimo](https://github.com/pacnimo/). All rights reserved.
     """) # Meta, project description, and footer added here
 
-gpt.launch(debug=True)
-
-st.title("Image Generation using API")
-
-# Streamlit app input fields
-prompt = st.text_input("Enter prompt:", "")
-negative_prompt = st.text_input("Enter negative prompt:", "")
-use_negative_prompt = st.checkbox("Use negative prompt", False)
-seed = st.number_input("Seed (0-2147483647):", value=0, min_value=0, max_value=2147483647)
-width = st.slider("Width:", 1024, 1536)
-height = st.slider("Height:", 1024, 1536)
-guidance_scale = st.slider("Guidance Scale:", 3.0, 20.0, 0.1)
-randomize_seed = st.checkbox("Randomize seed", True)
-
-if st.button("Generate Image"):
-    with st.spinner("Generating image..."):
-        img = generate_image(prompt, negative_prompt, use_negative_prompt, seed, width, height, guidance_scale, randomize_seed)
-        if img:
-            st.image(img, caption="Generated Image")
-        else:
-            st.error("The content retrieved is not an image.")
+app.launch(debug=True)