Update app.py

app.py

@@ -18,21 +18,14 @@ def get_webpage_text(url):
         return rawp
     else:
         return "ERROR couldn't find, "+url
-def generate_image(prompt):
-    client = InferenceClient("black-forest-labs/FLUX.1-dev")
-    response = client.text_to_image(prompt)
-    return response
 
-
-def generate_prompt(company_name, company_url=""):
-    
+def generate_prompt(company_name, company_html, company_descp):
     client = InferenceClient(model_name)
-    company_html = get_webpage_text(company_url)
-
     system_prompt=f"""You are a Master Generative Image Prompt Writer, you know just the perfect prompt secrets for every situation
     Today you will be generating Company Logo's
-    You will be given a Company Name, and HTML artifacts from their website, use this to generate a sufficiently long and detailed image generation prompt to satisfy the users request, make sure that the company name is the focal point of the image
+    You will be given a Company Name, Description, and HTML artifacts from their website, use this to generate a sufficiently long and detailed image generation prompt to satisfy the users request, make sure that the company name is the focal point of the image
     Company Name: {company_name}
+    Company Description: {company_descp}
     HTML from Company Website: {company_html}"""
     prompt_in=[
         {'role':'system','content':system_prompt},
@@ -43,19 +36,27 @@ def generate_prompt(company_name, company_url=""):
         output += response.token.text
     return output
 
+        
+def generate_image(comp_name, comp_desc, comp_url):
+    web_out=get_webpage_text(comp_url)
+    prompt_out=generate_prompt(comp_name,web_out, comp_desc)
+    client = InferenceClient("black-forest-labs/FLUX.1-dev")
+    response = client.text_to_image(prompt)
+    return response
+
 # Create Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## Smart Logo Maker")
     with gr.Row():
         with gr.Column():
-            model_dropdown = gr.Dropdown(models, label="Select Model")
-            prompt_input = gr.Textbox(label="Enter Company Name")
-            prompt_input = gr.Textbox(label="Enter Company Name")
+            comp_name = gr.Textbox(label="Enter Company Name")
+            comp_desc = gr.Textbox(label="Enter Company Description (optional)")
+            comp_url = gr.Textbox(label="Enter Company URL (optional)")
             generate_button = gr.Button("Generate Image")
         with gr.Column():
             output_image = gr.Image(label="Generated Image")
     
-    generate_button.click(generate_image, inputs=[prompt_input, model_dropdown], outputs=output_image)
+    generate_button.click(generate_image, inputs=[comp_name, comp_desc, comp_url], outputs=output_image)
 
 # Launch the interface
 demo.launch()