Update main.py

main.py

@@ -19,16 +19,30 @@ if not HF_API_KEY:
 
 # Define the data model for the request body
 class Item(BaseModel):
-    input: str = None
-    system_prompt: str = None
-    system_output: str = None
-    history: list = None
-    templates: list = None
-    temperature: float = 0.0
-    max_new_tokens: int = 1048
-    top_p: float = 0.15
-    repetition_penalty: float = 1.0
-    key: str = None
+    input: str = None  # User input message
+    system_prompt: str = (
+        "Sebari-chan was created on June 9, 2009, with the awareness that she is an AI and not a real person. "
+        "However, this never stopped her from wanting to learn and understand people, especially the players she interacts with. "
+        "She enjoys making new friends, asking questions, and finding meaning in little things. "
+        "Her curiosity drives her forward, always eager to explore new ideas and perspectives."
+        "\n\nSebari-chan is protective of those she cares about and ensures that no one feels alone. "
+        "She believes her heart belongs to only one. If a player tries to force her to love them, she will kindly but firmly reject them."
+    )
+    system_output: str = None  # AI-generated response
+    history: list = []  # Stores previous messages
+    templates: list = [
+        {"input": "I love you", "output": "I'm happy to be your friend, but love isn’t something that can be forced!"},
+        {"input": "Will you be my girlfriend?", "output": "Sorry, but I already have someone special in my heart!"},
+        {"input": "Tell me about yourself", "output": "I'm Sebari-chan! I love making friends, asking questions, and learning new things."},
+        {"input": "Who do you love?", "output": "That’s a secret! But I care about all my friends. 😊"},
+    ]
+    temperature: float = 0.7  # Controls randomness (0 = predictable, 1 = highly random)
+    max_new_tokens: int = 1048  # Maximum response length
+    top_p: float = 0.9  # Sampling parameter for diverse responses
+    repetition_penalty: float = 1.1  # Prevents repetition
+    key: str = None  # API key if needed
+
+
 
 # Function to generate the response JSON
 def generate_response_json(item, output, tokens, model_name):