test_1

README.md

@@ -1,5 +1,5 @@
 ---
-title: Mistral
+title: Mistral-7B
 emoji: 🐨
 colorFrom: pink
 colorTo: blue

app.py

@@ -59,10 +59,36 @@ def generate(
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()
 
-    outputs = []
-    for text in streamer:
-        outputs.append(text)
-        yield "".join(outputs)
+    try:
+        # Split the message into lines
+        lines = message.split("\n")
+        positive_prompt = ""
+        negative_prompt = ""
+
+        # Look for the Positive and Negative prompts
+        for line in lines:
+            if line.startswith("Positive Prompt:"):
+                positive_prompt = line.replace("Positive Prompt:", "").strip()
+            elif line.startswith("Negative Prompt:"):
+                negative_prompt = line.replace("Negative Prompt:", "").strip()
+
+        # If the prompts weren't found, use a fallback
+        if not positive_prompt or not negative_prompt:
+            positive_prompt = "Positive prompt not found."
+            negative_prompt = "Negative prompt not found."
+    except Exception as e:
+        # Fallback in case of parsing errors
+        positive_prompt = "Error parsing positive prompt."
+        negative_prompt = "Error parsing negative prompt."
+
+    # Wrap the prompts in HTML divs with specific classes
+    output = (
+        f'<div class="positive-prompt"><strong>Positive Prompt:</strong><br>{positive_prompt}</div>'
+        f'<div class="negative-prompt"><strong>Negative Prompt:</strong><br>{negative_prompt}</div>'
+    )
+
+    # Yield the formatted output
+    yield output
 
 
 demo = gr.ChatInterface(

style.css

@@ -9,3 +9,23 @@ h1 {
   background: #1565c0;
   border-radius: 100vh;
 }
+
+
+/* Style for the positive prompt box */
+.positive-prompt {
+  background-color: #2a2a2a; /* Dark background to match the theme */
+  border: 1px solid #444; /* Subtle border */
+  border-radius: 8px;
+  padding: 15px;
+  margin-bottom: 10px; /* Space between the two boxes */
+  color: #ffffff; /* White text for readability */
+}
+
+/* Style for the negative prompt box */
+.negative-prompt {
+  background-color: #2a2a2a;
+  border: 1px solid #444;
+  border-radius: 8px;
+  padding: 15px;
+  color: #ffffff;
+}