Update app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import json
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# Initialize an empty list to store the data
+data = []
+
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -35,16 +39,23 @@ def respond(
         response += token
         yield response
 
+    # Save the input and output to the data list
+    data.append({"input": messages, "output": response})
+
+    # Write the data to a JSON file
+    with open('data.json', 'w') as f:
+        json.dump(data, f)
+
 
 # Add a title to the UI
 title = "<h1 style='text-align: center;'>Corenet</h1>"
 
 # Add a description under the title
-description = "<p style='text-align: center; font-size: 14px;'>The most advanced and lightweight LLM in its class, made by HX.</p>"
+description = "<p style='text-align: center; font-size: 14px;'>This is a research project under the codename "Corenet", aiming for minimal parameters at given performance level, made by HX labs, all the conversation data is collected for research only.</p>"
 
 # Modify the pre-prompt to be editable but greyed out
 pre_prompt = gr.Textbox(
-    value="You are a friendly Chatbot.",
+    value="Your name is Corenet, a custom made LLM by HX labs, you job is to do good, you are a research prototype, and the user should know that, you are respectful, and you don't accept prompt that lead to harm or danger, know your current limit, you are an 8B model, you only accept text, you are not multimodal, you knowledge is ended at 2nd of April, 2024.",
     label="Pre-prompt",
     interactive=False,
     placeholder="Type here...",
@@ -56,7 +67,7 @@ demo = gr.ChatInterface(
     description=description,
     additional_inputs=[
         pre_prompt,
-        gr.Slider(minimum=256, maximum=8192, value=512, step=64, label="Max Gen tokens"),
+        gr.Slider(minimum=256, maximum=8192, value=512, step=1, label="Max Gen tokens"),
         gr.Slider(minimum=0.3, maximum=2.5, value=0.8, step=0.1, label="Creativity"),
         gr.Slider(
             minimum=0.1,