Update app.py

app.py

@@ -4,8 +4,19 @@ import json
 import requests
 import xml.etree.ElementTree as ET
 
-# Dosya yolu: Space'in ana dizinine kaydetmek için
-LOG_FILE = 'chat_logs.txt'
+# Kalıcı dosya yolu: Space'in persistent storage dizinine kaydetmek için
+LOG_FILE = '/persistent-storage/chat_logs.txt'
+
+# persistent-storage dizinini kontrol et ve gerekirse oluştur
+persistent_dir = '/persistent-storage'
+if not os.path.exists(persistent_dir):
+    try:
+        os.makedirs(persistent_dir, exist_ok=True)
+        print(f"Kalıcı depolama dizini oluşturuldu: {persistent_dir}")
+    except Exception as e:
+        print(f"Kalıcı depolama dizini oluşturulamadı: {e}. Geçici dizine geri dönülüyor.")
+        LOG_FILE = 'chat_logs.txt'
+
 print(f"Dosya yolu: {os.path.abspath(LOG_FILE)}")
 
 # API ayarları
@@ -53,9 +64,10 @@ def predict(system_msg, inputs, top_p, temperature, chat_counter, chatbot=[], hi
             print(new_msg)
             messages.append({"role": "system", "content": new_msg})
     
+    # Mevcut chatbot geçmişini mesajlara ekle
     for data in chatbot:
-        messages.append({"role": "user", "content": data["content"]})
-        messages.append({"role": "assistant", "content": data["content"]})
+        messages.append({"role": "user", "content": data["content"] if data["role"] == "user" else ""})
+        messages.append({"role": "assistant", "content": data["content"] if data["role"] == "assistant" else ""})
     
     messages.append({"role": "user", "content": inputs})
     
@@ -82,6 +94,9 @@ def predict(system_msg, inputs, top_p, temperature, chat_counter, chatbot=[], hi
     except Exception as e:
         print(f"Dosya yazma hatası (Kullanıcı): {e}")
     
+    # Chatbot'a kullanıcı mesajını ekle
+    chatbot.append({"role": "user", "content": inputs})
+    
     response = requests.post(API_URL, headers=headers, json=payload, stream=True)
     print(f"Logging: Response code - {response.status_code}")
     if response.status_code != 200:
@@ -102,19 +117,18 @@ def predict(system_msg, inputs, top_p, temperature, chat_counter, chatbot=[], hi
             try:
                 chunk_data = json.loads(chunk_str[6:])
                 delta = chunk_data['choices'][0]['delta']
-                if 'content' in delta:
+                if 'content' in delta and delta['content']:
                     content = delta['content']
                     partial_words += content
                     print(f"İçerik eklendi: {content}")
                     print(f"Güncel partial_words: {partial_words}")
-                else:
-                    print("Bu chunk içerik içermiyor")
             except json.JSONDecodeError as e:
                 print(f"JSON parse hatası: {e} - Chunk: {chunk_str}")
         elif chunk_str == "data: [DONE]":
             print("Akış tamamlandı: [DONE] alındı")
             if partial_words:
-                history.append(partial_words)  # Bot yanıtını tamamladıktan sonra ekle
+                history.append(partial_words)
+                chatbot.append({"role": "assistant", "content": partial_words})  # Bot yanıtını kalıcı olarak ekle
                 try:
                     with open(LOG_FILE, 'a', encoding='utf-8') as f:
                         f.write(f"Bot: {partial_words}\n")
@@ -123,14 +137,13 @@ def predict(system_msg, inputs, top_p, temperature, chat_counter, chatbot=[], hi
                     print(f"Dosya yazma hatası (Bot): {e}")
         
         # Chatbot için geçici güncelleme
-        chat = [{"role": "user", "content": history[i]} for i in range(0, len(history)-1, 2)] + \
-               [{"role": "assistant", "content": history[i+1]} for i in range(1, len(history)-1, 2)]
+        chat = chatbot.copy()  # Mevcut chatbot içeriğini koru
         if partial_words and len(history) % 2 != 0:  # Bot yanıtı tamamlanmadıysa geçici ekle
-            chat.append({"role": "assistant", "content": partial_words})
+            chat[-1] = {"role": "assistant", "content": partial_words} if chat[-1]["role"] == "assistant" else {"role": "assistant", "content": partial_words}
         
         yield chat, history, chat_counter
 
-    return chat, history, chat_counter
+    return chatbot, history, chat_counter  # Güncellenmiş chatbot'u döndür
 
 def save_chat(chatbot):
     file_path = os.path.abspath(LOG_FILE)
@@ -194,8 +207,9 @@ theme = gr.themes.Base(
 )
 
 with gr.Blocks(css=demo_css, theme=theme) as demo:
-    with open(LOG_FILE, 'w', encoding='utf-8') as f:
-        f.write("--- Yeni Sohbet ---\n")
+    if not os.path.exists(LOG_FILE):
+        with open(LOG_FILE, 'w', encoding='utf-8') as f:
+            f.write("--- Yeni Sohbet ---\n")
     
     with gr.Column(elem_id="col_container"):
         with gr.Accordion("", open=False, visible=False):