Update app.py

app.py

@@ -10,11 +10,23 @@ llm = Llama(
     use_mmap=True
 )
 
+def format_chat_history(history):
+    formatted_history = ""
+    for user_msg, assistant_msg in history:
+        formatted_history += f"USER: {user_msg}\nASSISTANT: {assistant_msg}\n"
+    return formatted_history
+
 def chat(message, history):
     system_prompt = """Ti si YugoGPT, AI asistent koji prvenstveno komunicira na srpskom jeziku.
-    Tvoj zadatak je da daješ detaljne i korisne odgovore na srpskom jeziku."""
+    Tvoj zadatak je da daješ detaljne i korisne odgovore na srpskom jeziku, uzimajući u obzir prethodnu konverzaciju."""
+    
+    chat_history = format_chat_history(history)
     
     full_prompt = f"""SYSTEM: {system_prompt}
+
+PRETHODNA KONVERZACIJA:
+{chat_history}
+
 USER: {message}
 ASSISTANT: """
     
@@ -38,11 +50,11 @@ ASSISTANT: """
 demo = gr.ChatInterface(
     fn=chat,
     title="YugoGPT Asistent",
-    description="Postavite pitanje na srpskom ili bilo kom jeziku bivše Jugoslavije.",
+    description="Vodite smislenu konverzaciju sa AI asistentom na srpskom jeziku.",
     examples=[
-        "Objasni kvantno računarstvo",
-        "Koje su glavne karakteristike veštačke inteligencije?",
-        "Kako funkcioniše blockchain tehnologija?"
+        "Kako si danas?",
+        "Šta misliš o veštačkoj inteligenciji?",
+        "Objasni mi blockchain tehnologiju"
     ]
 )
 
@@ -53,4 +65,3 @@ if __name__ == "__main__":
         share=False
     )
 
-