Switched from chatinterface to normal interface and has better use random seed option

app.py

@@ -16,8 +16,8 @@ else:
 model.to(device)
 
 def generate(
+    system_prompt,
     prompt,
-    history,
     max_new_tokens,
     repetition_penalty,
     temperature,
@@ -27,7 +27,7 @@ def generate(
     seed,
 ):
 
-    input_text = f"{prompt}, {history}"
+    input_text = f"{system_prompt}, {prompt}"
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
 
     if random_seed:
@@ -49,69 +49,29 @@ def generate(
     better_prompt = tokenizer.decode(outputs[0])
     return better_prompt
 
-use_random_seed = gr.Checkbox(value=False,  label="Use Random Seed", info="Check to use a random seed which is a start point for the generation process")
-
-
-additional_inputs = [
-    gr.Slider(
-        value=512,
-        minimum=250,
-        maximum=512,
-        step=1,
-        interactive=True,
-        label="Max New Tokens",
-        info="The maximum numbers of new tokens, controls how long is the output",
-    ),
-    gr.Slider(
-        value=1.2,
-        minimum=0,
-        maximum=2,
-        step=0.05,
-        interactive=True,
-        label="Repetition Penalty",
-        info="Penalize repeated tokens, making the AI repeat less itself",
-    ),
-    gr.Slider(
-        value=0.5,
-        minimum=0,
-        maximum=1,
-        step=0.05,
-        interactive=True,
-        label="Temperature",
-        info="Higher values produce more diverse outputs",
-    ),
-    gr.Slider(
-        value=1,
-        minimum=0,
-        maximum=2,
-        step=0.05,
-        interactive=True,
-        label="Top P",
-        info="Higher values sample more low-probability tokens",
-    ),
-    gr.Slider(
-        value=1,
-        minimum=1,
-        maximum=100,
-        step=1,
-        interactive=True,
-        label="Top K",
-        info="Higher k means more diverse outputs by considering a range of tokens",
-    ),
-    use_random_seed,
-    gr.Number(
-        value=42,
-        interactive=True,
-        label="Manual Seed",
-        info="A starting point to initiate the generation process",
-        visible={'False' if use_random_seed else 'True'}
-    ),
-]
+prompt = gr.Textbox(label="Prompt", interactive=True)
+
+system_prompt = gr.Textbox(label="System Prompt", interactive=True)
+
+max_new_tokens = gr.Slider(value=512, minimum=250, maximum=512, step=1, interactive=True, label="Max New Tokens", info="The maximum numbers of new tokens, controls how long is the output")
+    
+repetition_penalty = gr.Slider(value=1.2, minimum=0, maximum=2, step=0.05, interactive=True, label="Repetition Penalty", info="Penalize repeated tokens, making the AI repeat less itself")
+    
+temperature = gr.Slider(value=0.5, minimum=0, maximum=1, step=0.05, interactive=True, label="Temperature", info="Higher values produce more diverse outputs")
+
+top_p = gr.Slider(value=1, minimum=0, maximum=2, step=0.05, interactive=True, label="Top P", info="Higher values sample more low-probability tokens")
+
+top_k = gr.Slider(value=1, minimum=1, maximum=100, step=1, interactive=True, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens")
+
+use_random_seed = gr.Checkbox(value=False, label="Use Random Seed", info="Check to use a random seed which is a start point for the generation process")
+
+manual_seed = gr.Number(value=42, interactive=True, label="Manual Seed", info="A starting point to initiate the generation process", visible={'False' if use_random_seed else 'True'})
 
 
 examples = [
     [
-        "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it.",
+        "A storefront with 'Text to Image' written on it.",
+        "Expand the following prompt to add more detail:",
         512,
         1.2,
         0.5,
@@ -122,14 +82,13 @@ examples = [
     ]
 ]
 
-gr.ChatInterface(
+gr.Interface(
     fn=generate,
-    chatbot=gr.Chatbot(
-        show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"
-    ),
-    additional_inputs=additional_inputs,
+    inputs=[prompt, system_prompt, max_new_tokens, repetition_penalty, temperature, top_p, top_k, use_random_seed, manual_seed]
+    outputs=gr.Textbox(label="Better Prompt", interactive=True)
     title="SuperPrompt-v1",
     description="Make your prompts more detailed!",
     examples=examples,
+    live=True
     concurrency_limit=20,
 ).launch(show_api=False)