Update app.py

app.py

@@ -1,219 +1,50 @@
-# 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
-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
+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
         )
-        
-        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
+    return demo
 
-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]
-        )
-    
-    # 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.launch(
-        share=True,
-        show_error=True,
-        debug=True
-    )
+    demo = main()
+    demo.launch(share=True)
+