Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 # Your system prompt
-SYSTEM_PROMPT = "You are a prompt enhancer and your work is to enhance the given prompt under 100 words without changing the essence, only write the enhanced prompt and nothing else"
+SYSTEM_PROMPT = "Enhance and rewrite the prompt in sentence form without changing the essence"
 
 def format_prompt(message, history=[]):
     prompt = "<s>"
@@ -40,32 +40,12 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
         yield output.strip('</s>')
     return output.strip('</s>')
 
-def dark_theme():
-    return """
-        body {
-            background-color: #121212;
-            color: #ffffff;
-        }
-        .gradio-container {
-            background-color: #121212;
-            color: #ffffff;
-        }
-        input, textarea, select {
-            background-color: #333333 !important;
-            color: #ffffff !important;
-            border-color: #555555 !important;
-        }
-        .gradio-container button {
-            background-color: #555555 !important;
-            color: #ffffff !important;
-        }
-    """
+with gr.Blocks() as demo:
+    input_text = gr.Textbox(placeholder="Enter your prompt here...", lines=2, max_lines=2, label="Prompt")
+    submit_button = gr.Button("Generate")
+    output_text = gr.Textbox(label="Output", interactive=True, lines=10)
+    
+    
+    submit_button.click(fn=generate, inputs=input_text, outputs=output_text)
 
-with gr.Interface(
-    fn=generate,
-    inputs=gr.Textbox(placeholder="Enter your prompt here...", label="Prompt"),
-    outputs=gr.Textbox(label="Output", interactive=True, lines=10),
-    theme="compact",  # Use compact theme for default layout
-    css=dark_theme()  # Apply dark theme styles
-) as app:
-    app.launch() 
+demo.launch()