Add state management and localization

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import json
 import os
+import uuid
 
 ENDPOINT = os.environ.get("ENDPOINT", None)
 TOKEN = os.environ.get("TOKEN", None)
@@ -33,7 +34,8 @@ def convert_to_messages(message, history):
     return messages
 
 def chat_cosmosllama(message: str, 
-                     history: list
+                     history: list,
+                     conversation_id: str
 ) -> str:
     messages = convert_to_messages(message, history)
 
@@ -42,7 +44,7 @@ def chat_cosmosllama(message: str,
         "Content-Type": "application/json"
     }
 
-    response = requests.post(f"https://model.cosmosytu.online{ENDPOINT}", json={"messages": messages}, headers=headers, stream=True)
+    response = requests.post(f"https://model.cosmosytu.online{ENDPOINT}", json={"messages": messages, "conversation_id": conversation_id}, headers=headers, stream=True)
 
     if response.status_code == 200:
         full_response = ""
@@ -61,15 +63,34 @@ def chat_cosmosllama(message: str,
     else:
         yield "Bir hata oluştu."
 
+def get_conversation_id():
+    return str(uuid.uuid4())
+
+def reset_conversation():
+    new_conversation_id = get_conversation_id()
+    return new_conversation_id
+
 chatbot=gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='Cosmos LLaMa')
 
 with gr.Blocks(fill_height=True, css=css) as demo:
-    gr.ChatInterface(
+    
+    conversation_id_state = gr.State(value=get_conversation_id)
+    
+    interface = gr.ChatInterface(
         fn=chat_cosmosllama,
         chatbot=chatbot,
         fill_height=True,
         cache_examples=False,
+        additional_inputs=[conversation_id_state],
+        undo_btn=None,
+        retry_btn="🔄 Tekrar Cevapla",
+        clear_btn="🗑️ Temizle",
+        submit_btn="Gönder",
+        stop_btn="Durdur"
     )
 
+    interface.retry_btn.click(fn=reset_conversation, inputs=[], outputs=[conversation_id_state])
+    interface.textbox.placeholder="Mesajınızı buraya giriniz..."
+
 if __name__ == "__main__":
     demo.launch()