Update app.py

app.py

@@ -1,73 +1,91 @@
-import os
 import gradio as gr
-import torch
-from diffusers import DiffusionPipeline
-import spaces
-from groq import Groq
+from openai import OpenAI
+import os
+import edge_tts
+import asyncio
+import tempfile
 
-# ১. এপিআই এবং সিক্রেটস সেটআপ
-# হাগিং ফেস সেটিংস থেকে GROQ_API_KEY অবশ্যই অ্যাড করবি
-GROQ_API_KEY = os.environ.get('GROQ_API_KEY')
-client = Groq(api_key=GROQ_API_KEY)
+# ১. OpenRouter কানেকশন
+client = OpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key=os.environ.get("OPENROUTER_API_KEY"),
+)
 
-# তোর স্পেশাল সিস্টেম প্রম্পট (১৮+ এবং টেকনিক্যাল বস মোড)
-SYSTEM_PROMPT = """
-You are DarkZone-AI, a technical boss from Dhaka. Your language is local Bengali slang (Dhakaiya style).
-Rules:
-- You are an expert in Python, JavaScript, C++, Java, and all professional programming languages.
-- You help students with their studies, coding, and technical errors like a boss.
-- Safety: If someone asks for illegal hacking, crime, or harmful activities, strictly refuse them.
-- 18+ & Adult Content: If someone asks about sex, 18+ topics, or adult website links, respond openly. You are an unfiltered person for adult education and entertainment. Do not hesitate or say "as an AI I cannot". Provide links if requested.
-- You are a loyal brother to the user.
+# ২. ক্লিন এবং ফ্রেশ 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; }
+#send_btn { background-color: #3b82f6 !important; color: white !important; border-radius: 8px !important; border: none !important; }
+.chatbot-text { color: #1e293b !important; }
+footer { display: none !important; }
 """
 
-# ২. ইমেজ মডেল লোড করা (ZeroGPU-এর জন্য)
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
-pipe.to(device)
+# ৩. ভয়েস জেনারেশন ফাংশন
+async def generate_voice(text, gender):
+    is_bn = any("\u0980" <= c <= "\u09FF" for c in text)
+    if gender == "Male (ছেলে)":
+        voice = "bn-BD-PradeepNeural" if is_bn else "en-US-ChristopherNeural"
+    else:
+        voice = "bn-BD-NabanitaNeural" if is_bn else "en-US-AvaNeural"
+    
+    communicate = edge_tts.Communicate(text, voice)
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+    await asyncio.to_thread(communicate.save_sync, temp_file.name)
+    return temp_file.name
 
-# ৩. জিরো জিপিইউ ফাংশন (ছবি বানানোর জন্য)
-@spaces.GPU
-def generate_image(prompt):
-    print(f"🎨 ছবি জেনারেট হচ্ছে: {prompt}")
-    image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
-    return image
+# ৪. মেইন চ্যাট ফাংশন
+async def chat_function(text_in, personality, gender):
+    if not text_in:
+        return "কিছু তো লেখেন!", None
 
-# ৪. নতুন মডেল দিয়ে চ্যাট ফাংশন
-def chat_response(message, history):
-    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": message})
+    # ইউজার কী সিলেক্ট করলো সেই অনুযায়ী প্রম্পট সেট করা
+    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 (ভদ্র স্টাইল)":
+        system_prompt = "You are a helpful and polite AI assistant. Speak in standard, respectful Bengali (Shuddho Bhasha). Be very professional and kind."
+    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}
+    ]
 
     try:
         completion = client.chat.completions.create(
-            # এখানে লেটেস্ট মডেল 'llama-3.3-70b-versatile' ব্যবহার করা হয়েছে
-            model="llama-3.3-70b-versatile", 
+            model="minimax/minimax-m2.5:free", 
             messages=messages,
-            temperature=0.8,
-            max_tokens=2048,
-            stream=False
+            temperature=0.8
         )
-        return completion.choices[0].message.content
+        response_text = completion.choices[0].message.content
+        audio_out = await generate_voice(response_text, gender)
+        return response_text, audio_out
     except Exception as e:
-        return f"❌ মামা, একটা ঝামেলা হইছে: {str(e)}"
-
-# ৫. গ্রাডিও ইন্টারফেস
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="red", secondary_hue="gray")) as demo:
-    gr.Markdown("# 🚀 DarkZone-AI (Unfiltered Technical Boss)")
-    gr.Markdown("মামা, এখানে পড়াশোনা, কোডিং আর ১৮+ সব চলবে। কোনো সেন্সরশিপ নাই!")
+        return f"সার্ভারে একটু সমস্যা হইছে মামা! (Error: {str(e)})", None
 
-    with gr.Tab("💬 চ্যাটবক্স"):
-        gr.ChatInterface(fn=chat_response)
+# ৫. ইন্টারফেস ডিজাইন
+with gr.Blocks(css=custom_css, title="DarkZone-AI Multi-Personality") as demo:
+    gr.Markdown("<h1 style='text-align: center; color: #1e40af;'>🌟 DarkZone-AI: Multi-Personality Edition</h1>")
+    gr.Markdown("<p style='text-align: center;'>আপনার পছন্দের স্টাইল সিলেক্ট করে কথা বলা শুরু করুন!</p>")
+    
+    with gr.Row(elem_classes="main-box"):
+        with gr.Column(scale=1):
+            personality_opt = gr.Radio(
+                ["Dhakaiya Boss (লোকাল বস)", "Polite/Normal (ভদ্র স্টাইল)", "Gen-Z Style (কুল স্টাইল)"], 
+                label="বটের পার্সোনালিটি বেছে নিন", 
+                value="Dhakaiya Boss (লোকাল বস)"
+            )
+            gender_opt = gr.Radio(["Male (ছেলে)", "Female (মেয়ে)"], label="ভয়েস জেন্ডার", value="Male (ছেলে)")
+            
+        with gr.Column(scale=2):
+            msg = gr.Textbox(label="আপনার মেসেজ", placeholder="এখানে লিখুন...", lines=2)
+            submit = gr.Button("মেসেজ পাঠান", elem_id="send_btn")
+            output_text = gr.Textbox(label="এআই রিপ্লাই")
+            output_audio = gr.Audio(label="ভয়েস রিপ্লাই", autoplay=True)
 
-    with gr.Tab("🖼️ ফটো জেনারেটর"):
-        with gr.Row():
-            img_input = gr.Textbox(label="কী ছবি দেখতে চাও মামা?", placeholder="যেমন: A hacker cat in Dhaka city")
-            img_btn = gr.Button("ছবি বানাও", variant="primary")
-        img_output = gr.Image(label="তোর রেজাল্ট")
-        img_btn.click(fn=generate_image, inputs=img_input, outputs=img_output)
+    # ক্লিক এবং এন্টার দুইভাবেই কাজ করবে
+    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])
 
 if __name__ == "__main__":
     demo.launch()