Update app.py

app.py

@@ -9,7 +9,7 @@ from difflib import SequenceMatcher
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY", "")
 client = Groq(api_key=GROQ_API_KEY) if GROQ_API_KEY else None
 
-# --- ၁။ Dataset Loading (Safe Indexing) ---
+# --- ၁။ Dataset Loading ---
 chat_db = []
 try:
     csv_url = "https://huggingface.co/datasets/amkyawdev/AmkyawDev-Dataset/raw/main/train.csv"
@@ -17,7 +17,6 @@ try:
     if not df.empty:
         for _, row in df.iterrows():
             if len(row) >= 2:
-                # Column နာမည်တွေ လွဲနေရင်တောင် index နဲ့ ဆွဲထုတ်မယ်
                 u_text = str(row.iloc[0]).strip()
                 a_text = str(row.iloc[1]).strip()
                 if u_text != "nan" and a_text != "nan":
@@ -26,36 +25,31 @@ try:
 except Exception as e:
     print(f"⚠️ Dataset Warning: {e}")
 
-# --- ၂။ AI Logic (Mixtral-8x7b) ---
+# --- ၂။ Core AI Logic (Llama-3.1-8b-instant) ---
 def generate_response(user_input, history):
-    user_input = user_input.strip()
-
     # Step 1: Dataset Match
     for pair in chat_db:
-        if SequenceMatcher(None, user_input, pair["u"]).ratio() > 0.85:
+        if SequenceMatcher(None, user_input.lower(), pair["u"].lower()).ratio() > 0.85:
             res = pair["a"]
             for i in range(len(res)):
                 time.sleep(0.005)
                 yield res[: i + 1]
             return
 
-    # Step 2: Groq API
+    # Step 2: Groq API (Using Llama-3.1-8b-instant)
     if client:
         try:
-            # Mixtral 8x7b ကို သုံးထားတယ် (Natural ဖြစ်စေဖို့)
-            messages = [{"role": "system", "content": "You are a friendly Myanmar AI assistant."}]
-            
-            # History format ကို OpenAI style ပြောင်းပေးခြင်း
+            messages = [{"role": "system", "content": "You are Amkyaw AI, a helpful Myanmar assistant. Answer clearly in Unicode."}]
+            # Convert history to OpenAI format
             for h in history:
                 if h[0]: messages.append({"role": "user", "content": h[0]})
                 if h[1]: messages.append({"role": "assistant", "content": h[1]})
-            
             messages.append({"role": "user", "content": user_input})
 
             stream = client.chat.completions.create(
                 messages=messages,
-                model="llama-3.1-8b-instant", 
-                temperature=0.8,
+                model="llama-3.1-8b-instant", # ပြန်ပြောင်းလိုက်တဲ့ Model ID
+                temperature=0.7,
                 stream=True,
             )
             
@@ -65,61 +59,62 @@ def generate_response(user_input, history):
                     full_res += chunk.choices[0].delta.content
                     yield full_res
             return
-        except Exception:
-            pass
+        except Exception as e:
+            print(f"Groq Error: {e}")
 
-    yield "တောင်းပန်ပါတယ်၊ အဖြေရှာမတွေ့ပါဘူးခင်ဗျာ။"
+    yield "တောင်းပန်ပါတယ်၊ အခုလောလောဆယ် မဖြေနိုင်သေးပါဘူးခင်ဗျာ။"
 
-# --- ၃။ UI Design ---
+# --- ၃။ Professional Modern UI (CSS) ---
 custom_css = """
 footer { visibility: hidden !important; }
-.gradio-container { background: #0d1117 !important; color: white !important; }
+.gradio-container { background: #0d1117 !important; color: white !important; font-family: sans-serif; }
 .header-card {
     background: rgba(255, 255, 255, 0.03);
     border: 1px solid rgba(255, 255, 255, 0.1);
-    border-radius: 15px; padding: 15px; margin-bottom: 20px;
+    border-radius: 12px; padding: 15px 20px; margin-bottom: 20px;
 }
 #input-row {
-    background: #161b22 !important; border-radius: 30px !important;
+    background: #161b22 !important; border-radius: 25px !important;
     border: 1px solid #30363d !important; padding: 5px 15px !important;
+    box-shadow: 0 4px 10px rgba(0,0,0,0.2);
 }
 #input-box textarea { background: transparent !important; color: white !important; border: none !important; }
 #send-btn {
     background: #ffffff !important; color: #000000 !important;
-    border-radius: 50% !important; width: 42px !important; height: 42px !important;
+    border-radius: 50% !important; width: 40px !important; height: 40px !important; min-width: 40px !important;
+    font-weight: bold; cursor: pointer; transition: 0.2s;
 }
+#send-btn:hover { transform: scale(1.1); }
 """
 
 with gr.Blocks(title="Amkyaw AI v4") as demo:
     gr.HTML("""
         <div class="header-card">
             <div style="display:flex; justify-content: space-between; align-items: center;">
-                <div><b style="color: #58a6ff; font-size: 20px;">Amkyaw AI</b></div>
-                <div style="color:#8b949e; font-size:12px;">Mixtral-8x7B | Stable UI</div>
+                <b style="color: #58a6ff; font-size: 19px;">Amkyaw AI</b>
+                <span style="color:#8b949e; font-size:12px;">Llama 3.1 8B | Instant Speed</span>
             </div>
         </div>
     """)
     
-    # type="messages" ကို ဖြုတ်လိုက်ပါပြီ (TypeError ရှင်းဖို့)
-    chatbot = gr.Chatbot(show_label=False, height=600)
+    chatbot = gr.Chatbot(show_label=False, height=580)
     
     with gr.Row(elem_id="input-row"):
-        msg = gr.Textbox(placeholder="Ask me anything...", show_label=False, scale=15, elem_id="input-box", container=False)
+        msg = gr.Textbox(placeholder="Type your message...", show_label=False, scale=15, elem_id="input-box", container=False)
         submit = gr.Button("↑", elem_id="send-btn", scale=1)
 
     def user_flow(user_msg, history):
         if not user_msg: return "", history
-        # Format ဟောင်း [[user, bot]] အတိုင်း ပြန်သုံးပါတယ်
         return "", history + [[user_msg, None]]
 
     def bot_flow(history):
         user_input = history[-1][0]
-        history[-1][1] = ""
-        # generate_response ကို history အဟောင်းတွေပါ ပေးပို့တယ်
+        # Streaming response back to UI
         for chunk in generate_response(user_input, history[:-1]):
             history[-1][1] = chunk
             yield history
 
+    # Event sequence
     msg.submit(user_flow, [msg, chatbot], [msg, chatbot], queue=False).then(bot_flow, chatbot, chatbot)
     submit.click(user_flow, [msg, chatbot], [msg, chatbot], queue=False).then(bot_flow, chatbot, chatbot)