Update app.py

app.py

@@ -11,7 +11,7 @@ client = OpenAI(
     api_key=os.environ.get("OPENROUTER_API_KEY"),
 )
 
-# ২. ক্লিন এবং ফ্রেশ CSS (ডার্ক মুড বাদ দিয়ে সবার জন্য আরামদায়ক লুক)
+# ২. ক্লিন এবং ফ্রেশ CSS (আপনার অরিজিনাল CSS)
 custom_css = """
 .gradio-container { background-color: #f8fafc !important; font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; }
 .main-box { background: white !important; border-radius: 15px !important; box-shadow: 0 4px 6px -1px rgb(0 0 0 / 0.1); padding: 20px !important; }
@@ -20,7 +20,7 @@ custom_css = """
 footer { display: none !important; }
 """
 
-# ৩. ভয়েস জেনারেশন ফাংশন
+# ৩. ভয়েস জেনারেশন ফাংশন (অপরিবর্তিত)
 async def generate_voice(text, gender):
     is_bn = any("\u0980" <= c <= "\u09FF" for c in text)
     if gender == "Male (ছেলে)":
@@ -33,12 +33,12 @@ async def generate_voice(text, gender):
     await asyncio.to_thread(communicate.save_sync, temp_file.name)
     return temp_file.name
 
-# ৪. মেইন চ্যাট ফাংশন
-async def chat_function(text_in, personality, gender):
+# ৪. মেইন চ্যাট ফাংশন (হিস্ট্রি অ্যাড করা হয়েছে)
+async def chat_function(text_in, personality, gender, history):
     if not text_in:
-        return "কিছু তো লেখেন!", None
+        return "কিছু তো লেখেন!", None, history
 
-    # ইউজার কী সিলেক্ট করলো সেই অনুযায়ী প্রম্পট সেট করা
+    # পার্সোনালিটি সেট করা
     if personality == "Dhakaiya Boss (লোকাল বস)":
         system_prompt = "You are DarkZone-AI, a technical boss from Dhaka. Speak in local Dhakaiya Bengali slang. Be cool and a bit aggressive like a local big brother."
     elif personality == "Polite/Normal (ভদ্র স্টাইল)":
@@ -46,10 +46,16 @@ async def chat_function(text_in, personality, gender):
     else: # Gen-Z Style
         system_prompt = "You are a Gen-Z AI. Use trendy English-Bengali mix, slang like 'bro', 'lit', 'sus', 'vibe'. Be very energetic and informal."
 
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": text_in}
-    ]
+    # হিস্ট্রি ম্যানেজমেন্ট: মেসেজ লিস্ট তৈরি
+    messages = [{"role": "system", "content": system_prompt}]
+    
+    # আগের চ্যাট হিস্ট্রি মেসেজে যোগ করা
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    
+    # বর্তমান ইউজারের ইনপুট যোগ করা
+    messages.append({"role": "user", "content": text_in})
 
     try:
         completion = client.chat.completions.create(
@@ -58,13 +64,20 @@ async def chat_function(text_in, personality, gender):
             temperature=0.8
         )
         response_text = completion.choices[0].message.content
+        
+        # নতুন উত্তর হিস্ট্রিতে সেভ করা
+        history.append((text_in, response_text))
+        
         audio_out = await generate_voice(response_text, gender)
-        return response_text, audio_out
+        return response_text, audio_out, history
     except Exception as e:
-        return f"সার্ভারে একটু সমস্যা হইছে মামা! (Error: {str(e)})", None
+        return f"সার্ভারে একটু সমস্যা হইছে মামা! (Error: {str(e)})", None, history
 
 # ৫. ইন্টারফেস ডিজাইন
 with gr.Blocks(css=custom_css, title="DarkZone-AI Multi-Personality") as demo:
+    # চ্যাট হিস্ট্রি স্টোর করার জন্য স্টেট
+    history_state = gr.State([])
+    
     gr.Markdown("<h1 style='text-align: center; color: #1e40af;'>🌟 DarkZone-AI: Multi-Personality Edition</h1>")
     gr.Markdown("<p style='text-align: center;'>আপনার পছন্দের স্টাইল সিলেক্ট করে কথা বলা শুরু করুন!</p>")
     
@@ -83,9 +96,17 @@ with gr.Blocks(css=custom_css, title="DarkZone-AI Multi-Personality") as demo:
             output_text = gr.Textbox(label="এআই রিপ্লাই")
             output_audio = gr.Audio(label="ভয়েস রিপ্লাই", autoplay=True)
 
-    # ক্লিক এবং এন্টার দুইভাবেই কাজ করবে
-    submit.click(chat_function, inputs=[msg, personality_opt, gender_opt], outputs=[output_text, output_audio])
-    msg.submit(chat_function, inputs=[msg, personality_opt, gender_opt], outputs=[output_text, output_audio])
+    # ক্লিক এবং এন্টার হ্যান্ডেলিং (outputs-এ history_state যোগ করা হয়েছে)
+    submit.click(
+        chat_function, 
+        inputs=[msg, personality_opt, gender_opt, history_state], 
+        outputs=[output_text, output_audio, history_state]
+    )
+    msg.submit(
+        chat_function, 
+        inputs=[msg, personality_opt, gender_opt, history_state], 
+        outputs=[output_text, output_audio, history_state]
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()