Update app.py

app.py

@@ -1,49 +1,220 @@
+# import gradio as gr
+# import groq
+# from config.settings import settings
+# from core.rag_system import EnhancedRAGSystem
+# from core.tts_service import EnhancedTTSService
+# from core.wikipedia_processor import WikipediaProcessor
+# from services.audio_service import AudioService
+# from services.chat_service import ChatService
+# from services.image_service import ImageService
+# from services.streaming_voice_service import StreamingVoiceService
+# from ui.components import create_custom_css, create_header
+# from ui.tabs import create_all_tabs
+# def main():
+#     # Initialize clients and services
+#     if not settings.GROQ_API_KEY:
+#         raise ValueError("Please set the GROQ_API_KEY environment variable.")
+
+#     client = groq.Client(api_key=settings.GROQ_API_KEY)
+    
+#     # Initialize core systems
+#     rag_system = EnhancedRAGSystem()
+#     tts_service = EnhancedTTSService()
+#     wikipedia_processor = WikipediaProcessor()
+    
+#     # Initialize services
+#     audio_service = AudioService(client, rag_system, tts_service)
+#     chat_service = ChatService(client, rag_system, tts_service)
+#     image_service = ImageService(client)
+#     streaming_voice_service = StreamingVoiceService(client, rag_system, tts_service)
+    
+#     # Create Gradio interface
+#     with gr.Blocks(css=create_custom_css(), theme=gr.themes.Soft(primary_hue="orange", neutral_hue="slate")) as demo:
+#         create_header()
+#         gr.Markdown("### 🌐 Hệ thống Đa ngôn ngữ - Tự động chuyển đổi model theo ngôn ngữ")
+#         create_all_tabs(
+#             audio_service=audio_service,
+#             chat_service=chat_service,
+#             image_service=image_service,
+#             rag_system=rag_system,
+#             tts_service=tts_service,
+#             wikipedia_processor=wikipedia_processor,
+#             streaming_voice_service=streaming_voice_service
+#         )
+    
+#     return demo
+
+# if __name__ == "__main__":
+#     demo = main()
+#     demo.launch(share=True)
+
 import gradio as gr
+import os
 import groq
-from config.settings import settings
-from core.rag_system import EnhancedRAGSystem
-from core.tts_service import EnhancedTTSService
-from core.wikipedia_processor import WikipediaProcessor
-from services.audio_service import AudioService
-from services.chat_service import ChatService
-from services.image_service import ImageService
-from services.streaming_voice_service import StreamingVoiceService
-from ui.components import create_custom_css, create_header
-from ui.tabs import create_all_tabs
-def main():
-    # Initialize clients and services
-    if not settings.GROQ_API_KEY:
-        raise ValueError("Please set the GROQ_API_KEY environment variable.")
-
-    client = groq.Client(api_key=settings.GROQ_API_KEY)
-    
-    # Initialize core systems
-    rag_system = EnhancedRAGSystem()
-    tts_service = EnhancedTTSService()
-    wikipedia_processor = WikipediaProcessor()
-    
-    # Initialize services
-    audio_service = AudioService(client, rag_system, tts_service)
-    chat_service = ChatService(client, rag_system, tts_service)
-    image_service = ImageService(client)
-    streaming_voice_service = StreamingVoiceService(client, rag_system, tts_service)
-    
-    # Create Gradio interface
-    with gr.Blocks(css=create_custom_css(), theme=gr.themes.Soft(primary_hue="orange", neutral_hue="slate")) as demo:
-        create_header()
-        gr.Markdown("### 🌐 Hệ thống Đa ngôn ngữ - Tự động chuyển đổi model theo ngôn ngữ")
-        create_all_tabs(
-            audio_service=audio_service,
-            chat_service=chat_service,
-            image_service=image_service,
-            rag_system=rag_system,
-            tts_service=tts_service,
-            wikipedia_processor=wikipedia_processor,
-            streaming_voice_service=streaming_voice_service
+import traceback
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def test_system():
+    """Test toàn bộ hệ thống"""
+    try:
+        # Kiểm tra API Key
+        api_key = os.getenv("GROQ_API_KEY") or os.getenv("X_GROQ_API_KEY")
+        
+        if not api_key:
+            return "❌ LỖI: Không tìm thấy GROQ_API_KEY. Vui lòng thêm biến môi trường trong Hugging Face Spaces settings."
+        
+        # Test Groq client
+        client = groq.Client(api_key=api_key)
+        
+        # Test models
+        models = client.models.list()
+        available_models = [model.id for model in models.data]
+        
+        return f"✅ HỆ THỐNG HOẠT ĐỘNG\n• API Key: ***{api_key[-4:]}\n• Models: {len(available_models)} models\n• Ví dụ: {available_models[:3]}"
+        
+    except Exception as e:
+        return f"❌ LỖI HỆ THỐNG: {str(e)}\n\n💡 CÁCH SỬA:\n1. Kiểm tra GROQ_API_KEY trong Hugging Face Settings\n2. Đảm bảo API Key có định dạng: gsk_...\n3. Kiểm tra quota Groq"
+
+def simple_chat(message, history):
+    """Chat đơn giản với xử lý lỗi"""
+    if not message.strip():
+        return "", history
+    
+    try:
+        api_key = os.getenv("GROQ_API_KEY") or os.getenv("X_GROQ_API_KEY")
+        
+        if not api_key:
+            history.append((message, "❌ Lỗi: Không tìm thấy GROQ_API_KEY"))
+            return "", history
+        
+        client = groq.Client(api_key=api_key)
+        
+        response = client.chat.completions.create(
+            model="llama-3.1-8b-instant",
+            messages=[
+                {"role": "system", "content": "Bạn là trợ lý AI hữu ích. Trả lời ngắn gọn bằng tiếng Việt."},
+                {"role": "user", "content": message}
+            ],
+            max_tokens=150,
+            temperature=0.7
+        )
+        
+        bot_message = response.choices[0].message.content
+        history.append((message, bot_message))
+        
+    except Exception as e:
+        error_msg = f"❌ Lỗi: {str(e)}"
+        history.append((message, error_msg))
+    
+    return "", history
+
+def simple_audio_transcribe(audio):
+    """Transcribe audio đơn giản"""
+    try:
+        api_key = os.getenv("GROQ_API_KEY") or os.getenv("X_GROQ_API_KEY")
+        
+        if not api_key:
+            return "❌ Lỗi: Không tìm thấy GROQ_API_KEY", "Vui lòng kiểm tra cài đặt", "vi"
+        
+        if audio is None:
+            return "🎤 Vui lòng ghi âm hoặc upload file", "Tôi đang lắng nghe...", "vi"
+        
+        client = groq.Client(api_key=api_key)
+        
+        with open(audio, "rb") as audio_file:
+            transcription = client.audio.transcriptions.create(
+                model="whisper-large-v3-turbo",
+                file=audio_file,
+                response_format="text",
+                language="vi"
+            )
+        
+        # Simple response
+        response_client = client.chat.completions.create(
+            model="llama-3.1-8b-instant",
+            messages=[
+                {"role": "system", "content": "Bạn là trợ lý AI. Trả lời ngắn gọn bằng tiếng Việt."},
+                {"role": "user", "content": transcription}
+            ],
+            max_tokens=100
+        )
+        
+        response = response_client.choices[0].message.content
+        
+        return transcription, response, "vi"
+        
+    except Exception as e:
+        error_msg = f"❌ Lỗi: {str(e)}"
+        return error_msg, error_msg, "vi"
+
+# Tạo interface đơn giản
+with gr.Blocks(theme=gr.themes.Soft(), title="AI Assistant - Hugging Face") as demo:
+    gr.Markdown("# 🤖 AI Assistant - Hugging Face Spaces")
+    gr.Markdown("Phiên bản đơn giản hoạt động trên Hugging Face")
+    
+    # System Test
+    with gr.Row():
+        test_btn = gr.Button("🔍 Test Hệ Thống", variant="secondary")
+        test_output = gr.Textbox(label="Kết Quả Test", interactive=False)
+    
+    test_btn.click(test_system, outputs=[test_output])
+    
+    # Simple Chat
+    with gr.Tab("💬 Chat Đơn Giản"):
+        chatbot = gr.Chatbot(height=400, label="Cuộc Trò Chuyện")
+        msg = gr.Textbox(placeholder="Nhập tin nhắn của bạn...", label="Tin Nhắn")
+        clear_btn = gr.Button("🗑️ Xóa Lịch Sử")
+        
+        def clear_chat():
+            return []
+        
+        msg.submit(simple_chat, [msg, chatbot], [msg, chatbot])
+        clear_btn.click(clear_chat, outputs=[chatbot])
+    
+    # Simple Audio
+    with gr.Tab("🎙️ Ghi Âm Đơn Giản"):
+        audio_input = gr.Audio(
+            sources=["microphone", "upload"],
+            type="filepath",
+            label="Ghi Âm hoặc Upload File"
+        )
+        
+        with gr.Row():
+            transcription = gr.Textbox(label="Bản Ghi Âm", lines=3)
+            response = gr.Textbox(label="Phản Hồi AI", lines=3)
+            language = gr.Textbox(label="Ngôn Ngữ", value="vi")
+        
+        audio_btn = gr.Button("🎯 Xử Lý Âm Thanh", variant="primary")
+        
+        audio_btn.click(
+            simple_audio_transcribe,
+            inputs=[audio_input],
+            outputs=[transcription, response, language]
         )
     
-    return demo
+    # Instructions
+    with gr.Accordion("📖 Hướng Dẫn Sử Dụng", open=False):
+        gr.Markdown("""
+        ## 🔧 Cài Đặt Trên Hugging Face Spaces
+        
+        1. **Thêm GROQ_API_KEY:**
+           - Vào Settings → Repository secrets
+           - Thêm secret: `GROQ_API_KEY` với giá trị API key từ Groq
+        
+        2. **API Key Format:**
+           - Phải bắt đầu bằng `gsk_`
+           - Lấy từ: https://console.groq.com
+        
+        3. **Nếu vẫn lỗi:**
+           - Kiểm tra quota Groq API
+           - Đảm bảo API key có quyền truy cập
+        """)
 
 if __name__ == "__main__":
-    demo = main()
-    demo.launch(share=True)
+    demo.launch(
+        share=True,
+        show_error=True,
+        debug=True
+    )