Update app.py

app.py

@@ -21,6 +21,7 @@ class PromptRequest(BaseModel):
     prompt: str
     max_new_tokens: int = 300
     temperature: float = 0.7
+    history: Optional[List[str]] = []
 
 @app.post("/generate")
 async def generate_text(req: PromptRequest):
@@ -43,9 +44,15 @@ async def generate_text(req: PromptRequest):
 
     **Gaya Bahasa**: Gunakan bahasa yang edukatif, memotivasi, dan mudah dipahami.
 """
-        messages = [f"<<SYS>>\n{system_prompt}\n<</SYS>>"]
-        messages.extend(req.history)  # Tambahkan history jika ada
-        messages.append(req.prompt)   #
+        messages = [
+            Content(role="user", parts=[{"text": f"<<SYS>>\n{system_prompt}\n<</SYS>>"}])
+        ]
+        for msg in req.history or []:
+            role = "user" if msg.startswith("User:") else "model"
+            text = msg.replace("User: ", "").replace("Bot: ", "")
+            messages.append(Content(role=role, parts=[{"text": text}]))
+
+        messages.append(Content(role="user", parts=[{"text": req.prompt}]))
         result = model.generate_content(
             messages,
             generation_config={