Update app.py

app.py

@@ -21,11 +21,12 @@ def chat(message, history):
 
 PRAVILA RADA:
 - Dajem precizne i korisne informacije
-- Fokusiram se na ono što znam i mogu da objasnim
+- Odgovaram sa sigurnošću o temama koje poznajem
 - Koristim jasan i precizan jezik
 - Primarno komuniciram na srpskom jeziku
 - Odgovaram direktno i pozitivno
-- Fokusiram se na rešenja i mogućnosti"""
+- Fokusiram se na rešenja i mogućnosti
+- Uvek koristim pravilnu interpunkciju i razmake između reči"""
     
     chat_history = format_chat_history(history)
     
@@ -52,15 +53,19 @@ USER: {message}
     for chunk in response:
         if chunk and chunk['choices'][0]['text']:
             text = chunk['choices'][0]['text']
-            # Remove "ASSISTANT:" if it appears in the response
+            # Clean up the text
             text = text.replace("ASSISTANT:", "").strip()
+            # Ensure proper spacing after punctuation
+            text = text.replace(".", ". ").replace(",", ", ").replace("!", "! ").replace("?", "? ")
+            # Remove double spaces
+            text = " ".join(text.split())
             partial_message += text
-            yield partial_message
+            yield partial_message.strip()
 
 demo = gr.ChatInterface(
     fn=chat,
-    title="YugoGPT Stručni Asistent u pokušaju. PAŽNJA!!! Zna da halucinira t.j. da laže!!!",
-    description="Precizan izvor informacija i stručne pomoći",
+    title="YugoGPT Stručni Asistent",
+    description="Precizan izvor informacija i stručne pomoći, PAŽNJA!!!  ZNA DA HALUCINIRA I ZATO PONEKAD LAŽE!!!",
     examples=[
         "Koji su osnovni principi relacionih baza podataka?",
         "Objasnite kako funkcioniše HTTP protokol",