update

app.py

@@ -20,7 +20,7 @@ st.set_page_config(
 # --- 2. CẤU HÌNH BACKEND ---
 API_URL = "https://Vanhwbt-TraPhuHoi-Backend.hf.space/predict"
 
-# --- 3. HÀM LƯU DỮ LIỆU ĐA NĂNG (HF + IMGBB + SHEET) ---
+# --- 2.5. HÀM LƯU DỮ LIỆU ĐA NĂNG (HF + IMGBB + SHEET) ---
 import gspread
 import json
 import io
@@ -180,41 +180,51 @@ disease_info = {
     }
 }
 
-# --- Cấu hình Gemini (Phiên bản Bất Tử - Fix mọi lỗi Key) ---
+# --- Cấu hình Gemini (Phiên bản Bất Tử - Fix mọi lỗi Key & Quota) ---
+import random # <--- Nhớ thêm thư viện này để bốc thăm
+
 try:
-    # 1. Kiểm tra xem có key 'TraPhuHoi' trong secrets không
-    if "TraPhuHoi" in st.secrets:
-        secret_value = st.secrets["TraPhuHoi"]
-        
-        # 2. Xử lý thông minh: 
-        # Nếu Hugging Face trả về string (chỉ có mã key AIzaSy...) -> Dùng luôn
-        if isinstance(secret_value, str):
-            GEMINI_API_KEY = secret_value
-        
-        # Nếu trả về Dictionary (kiểu TOML: [TraPhuHoi] api_key = "...") -> Lấy field api_key
-        elif isinstance(secret_value, dict) and "api_key" in secret_value:
-            GEMINI_API_KEY = secret_value["api_key"]
+    # 1. Gom tất cả các Key vào một danh sách
+    potential_keys = ["TraPhuHoi", "TraPhuHoi1", "TraPhuHoi2", "TraPhuHoi3"]
+    available_keys = []
+
+    for key_name in potential_keys:
+        if key_name in st.secrets:
+            secret_value = st.secrets[key_name]
             
-        # Nếu không đúng cả 2
-        else:
-            GEMINI_API_KEY = None
+            # Xử lý: Lấy chuỗi key sạch (dù là String hay Dict)
+            real_key = None
+            if isinstance(secret_value, str):
+                real_key = secret_value
+            elif isinstance(secret_value, dict) and "api_key" in secret_value:
+                real_key = secret_value["api_key"]
             
-        # 3. Tiến hành cấu hình
-        if GEMINI_API_KEY:
-            genai.configure(api_key=GEMINI_API_KEY)
-            model_gemini = genai.GenerativeModel('gemini-2.5-flash')
-            gemini_configured = True
-        else:
-            gemini_configured = False
-            model_gemini = None
+            # Nếu lấy được key thì bỏ vào rổ
+            if real_key:
+                available_keys.append(real_key)
+
+    # 2. Bốc thăm và Cấu hình
+    if available_keys:
+        # Chọn ngẫu nhiên 1 key trong danh sách
+        chosen_key = random.choice(available_keys)
+        
+        # In log ra terminal để bro biết đang dùng key nào (Debug)
+        print(f"🔑 Đang sử dụng Key (đuôi): ...{chosen_key[-6:]}")
+        
+        genai.configure(api_key=chosen_key)
+        
+        # [QUAN TRỌNG] Dùng bản 1.5 Flash để tránh lỗi 400 và tiết kiệm
+        model_gemini = genai.GenerativeModel('gemini-1.5-flash')
+        gemini_configured = True
     else:
-        # Không tìm thấy secret tên TraPhuHoi
+        # Không tìm thấy bất kỳ key nào
+        print("❌ Không tìm thấy key Gemini nào trong Secrets!")
         gemini_configured = False
         model_gemini = None
 
 except Exception as e:
     # Log lỗi ra terminal server để debug nếu cần
-    print(f"Lỗi cấu hình Gemini: {e}")
+    print(f"🔥 Lỗi cấu hình Gemini: {e}")
     gemini_configured = False
     model_gemini = None
 
@@ -243,69 +253,82 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
-# ================== 🤖 CHATBOT ==================
-st.divider()
-st.subheader("🤖 Trò chuyện cùng Chuyên Gia Trà Phú Hội")
-
-if gemini_configured and model_gemini:
-        # [FIX] Sửa prompt thành Cây Trà
-        personality_prompt = """
-        Tôi là Bác sĩ AgriDoc, chuyên gia về CÂY TRÀ PHÚ HỘI.
-        Mục tiêu: Hỗ trợ bà con chẩn đoán bệnh trên cây trà.
-        Phong cách: Thân thiện, ngắn gọn, thực tế.
-        Quy tắc: Chỉ trả lời về cây trà. Dựa vào dữ liệu đã học.
-        """
-        knowledge_base = "\n\n# DỮ LIỆU BỆNH:\n"
-        for key, info in disease_info.items():
-            knowledge_base += f"- {info['ten']}: {info['mo_ta']} | Xử lý: {', '.join(info['xu_ly'])}\n"
-
-        initial_context = personality_prompt + knowledge_base
-        initial_greeting = "Chào bà con! Tui là chuyên gia cây trà đây, bà con cần hỏi bệnh gì?"
+# ================== 🤖 CHATBOT (POWERED BY GEMMA 9B) ==================
+from huggingface_hub import InferenceClient # Nhớ cài thư viện: pip install huggingface_hub
 
-        # Nút xóa hội thoại
-        if st.button("🗑️ Xóa hội thoại"):
-            st.session_state.clear()
-            st.rerun()
+st.divider()
+st.subheader("🤖 Trò chuyện cùng Chuyên Gia Trà Phú Hội (Gemma AI)")
 
-        if "chat_history_gemini" not in st.session_state:
-            st.session_state.chat_history_gemini = [{"role": "assistant", "content": initial_greeting}]
+# --- 1. HÀM GỌI GEMMA 9B ---
+def chat_with_gemma(prompt):
+    try:
+        # Kiểm tra Token
+        if "HF_TOKEN" not in st.secrets:
+            return "⚠️ Lỗi: Chưa có HF_TOKEN trong Secrets."
         
-        if "gemini_chat" not in st.session_state:
-            try:
-                st.session_state.gemini_chat = model_gemini.start_chat(history=[
-                    {'role': 'user', 'parts': [initial_context]},
-                    {'role': 'model', 'parts': [initial_greeting]}
-                ])
-            except:
-                 st.error("Lỗi khởi tạo Chat.")
-                 st.stop()
-
-        # Hiện lịch sử chat
-        for message in st.session_state.chat_history_gemini:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
-
-        # Nhập câu hỏi
-        if user_input := st.chat_input("Hỏi về bệnh trà..."):
-            st.session_state.chat_history_gemini.append({"role": "user", "content": user_input})
-            with st.chat_message("user"):
-                st.markdown(user_input)
-
-            try:
-                with st.spinner("Đang suy nghĩ..."):
-                    response = st.session_state.gemini_chat.send_message(user_input, stream=True)
-                    with st.chat_message("assistant"):
-                        response_text = ""
-                        response_placeholder = st.empty()
-                        for chunk in response:
-                            response_text += chunk.text
-                            response_placeholder.markdown(response_text + "▌")
-                        response_placeholder.markdown(response_text)
-                    st.session_state.chat_history_gemini.append({"role": "assistant", "content": response_text})
-            except Exception as e:
-                st.error(f"Lỗi API: {e}")
-else:
-    st.warning("⚠️ Chưa kết nối được Gemini (Kiểm tra lại Secrets).")
+        # Khởi tạo Client
+        client = InferenceClient(api_key=st.secrets["HF_TOKEN"])
+        
+        # Thiết lập tính cách chuyên gia
+        system_instruction = """
+        Bạn là Kỹ sư Nông nghiệp chuyên về Cây Trà Phú Hội (Nhơn Trạch, Đồng Nai).
+        Nhiệm vụ: Tư vấn kỹ thuật trồng, chăm sóc, trị bệnh (Rầy nhảy, Đốm rong, Cháy lá...) cho bà con.
+        Phong cách: Thân thiện, ngắn gọn, súc tích, dùng từ ngữ bình dân dễ hiểu.
+        Tuyệt đối: Trả lời Tiếng Việt. Nếu không biết thì khuyên bà con hỏi cán bộ khuyến nông.
+        """
+        
+        # Tạo tin nhắn gửi đi
+        messages = [
+            {"role": "user", "content": f"{system_instruction}\n\nCâu hỏi của bà con: {prompt}"}
+        ]
+        
+        # Gọi Model Gemma-2-9B (Bản thông minh, miễn phí)
+        response = client.chat_completion(
+            model="google/gemma-2-9b-it", 
+            messages=messages, 
+            max_tokens=800, 
+            stream=False # Tắt stream để chạy ổn định hơn trên Web
+        )
+        return response.choices[0].message.content
+        
+    except Exception as e:
+        return f"😔 Kỹ sư Gemma đang bận (Lỗi kết nối): {e}"
+
+# --- 2. GIAO DIỆN CHAT ---
+
+# Nút xóa hội thoại
+if st.button("🗑️ Xóa hội thoại"):
+    st.session_state.chat_history_gemini = []
+    st.rerun()
+
+# Khởi tạo lịch sử nếu chưa có
+if "chat_history_gemini" not in st.session_state:
+    st.session_state.chat_history_gemini = [
+        {"role": "assistant", "content": "Chào bà con! Tui là Kỹ sư Gemma (Trí tuệ nhân tạo), bà con cần hỏi bệnh gì về cây trà không?"}
+    ]
+
+# Hiện lịch sử chat cũ
+for message in st.session_state.chat_history_gemini:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# --- 3. XỬ LÝ KHI NGƯỜI DÙNG NHẬP CÂU HỎI ---
+if user_input := st.chat_input("Hỏi về bệnh trà, cách bón phân..."):
+    # A. Hiện câu hỏi của người dùng ngay lập tức
+    st.session_state.chat_history_gemini.append({"role": "user", "content": user_input})
+    with st.chat_message("user"):
+        st.markdown(user_input)
+
+    # B. Gọi Gemma trả lời
+    with st.spinner("Kỹ sư Gemma đang suy nghĩ..."):
+        bot_reply = chat_with_gemma(user_input)
+        
+        # Hiện câu trả lời của Bot
+        with st.chat_message("assistant"):
+            st.markdown(bot_reply)
+        
+        # Lưu vào lịch sử
+        st.session_state.chat_history_gemini.append({"role": "assistant", "content": bot_reply})
 
 # --- 5. GIAO DIỆN CHÍNH (HTML) ---
 st.markdown("""

requirements.txt

@@ -4,4 +4,5 @@ pillow
 google-generativeai
 gspread
 oauth2client
-google-api-python-client
+google-api-python-client
+huggingface_hub