Update app.py

app.py

@@ -1,17 +1,17 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
+# Initialize the HuggingFace Inference Client with the specified model
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
 
-def format_prompt(message, job_profession):
-    system_prompt = f"You are a People Generator. You generate profiles based on user input. Please ensure that all responses are written in the second-person perspective. For example, a response should be structured as follows: 'You are (Name), (Age) years old from (Country), and an expert in the field of {job_profession}.' This format must be strictly followed in all outputs."
-    prompt = f"<s>[SYS] {system_prompt} [/SYS][INST] {message} [/INST]</s>"
+def format_prompt(message, logo_request):
+    system_prompt = """
+    You are an advanced language model designed to create detailed and creative image prompts for logo generation. Based on the user's input, generate an elaborate and descriptive image prompt that can be used to create a high-quality logo. Ensure that the prompt is clear, imaginative, and provides specific details that will guide the logo creation process effectively.
+    """
+    prompt = f"<s>[SYS] {system_prompt} [/SYS][INST] {logo_request} [/INST]</s>"
     return prompt
 
-def generate(message, job_profession, temperature=0.9, max_new_tokens=4192, top_p=0.95, repetition_penalty=1.0):
-    # Parse the input to determine job profession based on the presence of a message.
-    actual_job = message if message else job_profession
-
+def generate(logo_request, temperature=0.9, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -26,7 +26,7 @@ def generate(message, job_profession, temperature=0.9, max_new_tokens=4192, top_
         "seed": 42,
     }
 
-    formatted_prompt = format_prompt("", actual_job)  # message should be empty here if we're using job from input
+    formatted_prompt = format_prompt("", logo_request)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
@@ -46,30 +46,25 @@ with gr.Blocks(css=css) as gpt:
     with gr.Row():
         with gr.Column(scale=2):
             gr.HTML("<h1>Settings</h1>")
-            job_dropdown = gr.Dropdown(
-                label="Choose a job profession",
-                choices=["Relationship Expert", "Sales Manager", "Blue Team Hacker", "Senior Web Developer", "C++ Developer", "Article Author", "News Anchor", "Finance Advisor"],
-                value="Relationship Expert"  # Initial value
-            )
+            logo_input = gr.Textbox(label="Input your logo request", placeholder="Describe the logo you want...")
         with gr.Column(scale=3):
-            gr.HTML("<h1><center>GPT Prompt Generator<h1><center>")
-            message_input = gr.Textbox(label="Input your message (overrides job dropdown)", placeholder="Type a job profession here to override the dropdown...")
+            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)
             generate_button.click(
                 fn=generate,
-                inputs=[message_input, job_dropdown],
+                inputs=[logo_input],
                 outputs=output_area
             )
 
     gr.Markdown("""
     ---
     ### Meta Information
-    **Project Title**: GPT Prompt Generator
+    **Project Title**: Logo Prompt Generator
     
-    **Github**: [https://github.com/pacnimo/gpt-prompt-generator](https://github.com/pacnimo/gpt-prompt-generator)
+    **Github**: [https://github.com/pacnimo/gpt-prompt-generator](https://github.com/pacnimo/)
     
-    **Description**: GPT Prompt Generator is Free and Easy to Use. Create a GPT Prompt Based on the Profession. 1 Click Prompt Generator.
+    **Description**: Logo Prompt Generator is Free and Easy to Use. Create a GPT Prompt Based on the Logo Request. 1 Click Prompt Generator.
     
     **Footer**: © 2024 by [pacnimo](https://github.com/pacnimo/). All rights reserved.
     """) # Meta, project description, and footer added here