Revert lại bản UI cũ

app.py

@@ -1,6 +1,5 @@
 """
 Chainlit UI for AI Agent with Login/Registration and Image Support
-Fixed version with proper timeout handling
 """
 
 import chainlit as cl
@@ -8,68 +7,25 @@ import requests
 from typing import Optional, Dict
 import json
 import os
-import asyncio
 from image_captioning import image_captioning
 
 # API Configuration
 API_BASE_URL = os.getenv("API_BASE_URL")
 
-# Timeout configurations
-API_TIMEOUT = 180  # 3 minutes for API calls
-IMAGE_PROCESSING_TIMEOUT =150  # 2.5 minutes for image processing
-
 # ============================================
 # API HELPER FUNCTIONS
 # ============================================
 
-def check_user_exists(user_id: str) -> dict:
-    """Check if user exists in the system - returns dict with status and message"""
+def check_user_exists(user_id: str) -> bool:
+    """Check if user exists in the system"""
     try:
-        url = f"{API_BASE_URL}/user-exists/{user_id}"
-        print(f"[DEBUG] Checking user exists at: {url}")
-        
-        response = requests.get(url, timeout=10)
-        print(f"[DEBUG] Response status: {response.status_code}")
-        print(f"[DEBUG] Response body: {response.text}")
-        
+        response = requests.get(f"{API_BASE_URL}/user-exists/{user_id}")
         if response.status_code == 200:
-            data = response.json()
-            exists = data.get("exists", False)
-            print(f"[DEBUG] User exists: {exists}, Full data: {data}")
-            return {
-                "success": True,
-                "exists": exists,
-                "user_id": data.get("user_id", user_id),
-                "message": data.get("message", "")
-            }
-        else:
-            print(f"[ERROR] Unexpected status code: {response.status_code}")
-            return {
-                "success": False,
-                "exists": False,
-                "error": f"API returned status {response.status_code}"
-            }
-    except requests.exceptions.Timeout:
-        print(f"[ERROR] Timeout when checking user: {user_id}")
-        return {
-            "success": False,
-            "exists": False,
-            "error": "Timeout - API không phản hồi"
-        }
-    except requests.exceptions.ConnectionError as e:
-        print(f"[ERROR] Connection error: {e}")
-        return {
-            "success": False,
-            "exists": False,
-            "error": f"Lỗi kết nối: {str(e)}"
-        }
+            return response.json().get("exists", False)
+        return False
     except Exception as e:
-        print(f"[ERROR] Error checking user: {type(e).__name__} - {e}")
-        return {
-            "success": False,
-            "exists": False,
-            "error": f"Lỗi: {str(e)}"
-        }
+        print(f"Error checking user: {e}")
+        return False
 
 
 def register_user(user_id: str) -> Dict:
@@ -77,8 +33,7 @@ def register_user(user_id: str) -> Dict:
     try:
         response = requests.post(
             f"{API_BASE_URL}/register",
-            json={"user_id": user_id},
-            timeout=10
+            json={"user_id": user_id}
         )
         if response.status_code == 201:
             return {"success": True, "message": response.json().get("message", "Đăng ký thành công!")}
@@ -89,20 +44,14 @@ def register_user(user_id: str) -> Dict:
         return {"success": False, "message": f"Lỗi kết nối: {str(e)}"}
 
 
-async def send_chat_message_async(query: str, user_id: str) -> Dict:
-    """Send chat message to API with async support"""
+def send_chat_message(query: str, user_id: str) -> Dict:
+    """Send chat message to API"""
     try:
-        # Run the blocking request in a thread pool
-        loop = asyncio.get_event_loop()
-        response = await loop.run_in_executor(
-            None,
-            lambda: requests.post(
-                f"{API_BASE_URL}/chat",
-                json={"query": query, "user_id": user_id},
-                timeout=API_TIMEOUT
-            )
+        response = requests.post(
+            f"{API_BASE_URL}/chat",
+            json={"query": query, "user_id": user_id},
+            timeout=120
         )
-        
         if response.status_code == 200:
             data = response.json()
             return {
@@ -114,8 +63,6 @@ async def send_chat_message_async(query: str, user_id: str) -> Dict:
             error_msg = response.json().get("detail", "Lỗi xử lý câu hỏi")
             return {"success": False, "message": error_msg}
             
-    except asyncio.TimeoutError:
-        return {"success": False, "message": "⏱️ Request timeout! Vui lòng thử lại."}
     except requests.exceptions.Timeout:
         return {"success": False, "message": "⏱️ Server đang khởi động (cold start), vui lòng thử lại sau 30 giây!"}
     except Exception as e:
@@ -126,18 +73,11 @@ async def send_chat_message_async(query: str, user_id: str) -> Dict:
 # IMAGE PROCESSING
 # ============================================
 
-async def process_image_async(image_path: str) -> str:
-    """Process image and return caption with async support"""
+async def process_image(image_path: str) -> str:
+    """Process image and return caption"""
     try:
-        # Run the blocking image_captioning in a thread pool
-        loop = asyncio.get_event_loop()
-        caption = await asyncio.wait_for(
-            loop.run_in_executor(None, image_captioning, image_path),
-            timeout=IMAGE_PROCESSING_TIMEOUT
-        )
+        caption = image_captioning(image_path)
         return caption
-    except asyncio.TimeoutError:
-        return "(⏱️ Timeout khi xử lý ảnh - ảnh quá phức tạp hoặc lỗi xử lý)"
     except Exception as e:
         return f"(Lỗi khi xử lý ảnh: {str(e)})"
 
@@ -211,7 +151,7 @@ async def main(message: cl.Message):
     user_id = cl.user_session.get("user_id")
     awaiting_auth = cl.user_session.get("awaiting_auth", True)
     
-    content = message.content.strip() if message.content else ""
+    content = message.content.strip()
     
     # Handle authentication commands
     if content.startswith("/"):
@@ -239,21 +179,28 @@ async def process_chat_message_with_image(message: cl.Message, user_id: str):
     
     # If there are images, process them
     if images:
+        # Show processing message
+        processing_msg = cl.Message(content="🖼️ Đang xử lý ảnh...", author="Kumiko")
+        await processing_msg.send()
+        
         image_captions = []
         
-        for idx, img in enumerate(images, 1):
+        for img in images:
             try:
                 # Get image path
                 image_path = img.path
                 
-                # Process image and get caption with async
-                caption = await process_image_async(image_path)
-                image_captions.append(f"[Ảnh {idx}]: {caption}")
+                # Process image and get caption
+                caption = await process_image(image_path)
+                image_captions.append(caption)
                 
             except Exception as e:
-                error_caption = f"[Ảnh {idx}]: (Lỗi khi xử lý ảnh: {str(e)})"
+                error_caption = f"(Lỗi khi xử lý ảnh: {str(e)})"
                 image_captions.append(error_caption)
         
+        # Remove processing message
+        await processing_msg.remove()
+        
         # Concatenate all captions
         combined_captions = "\n\n".join(image_captions)
         
@@ -299,39 +246,14 @@ async def handle_command(command: str):
         loading_msg = cl.Message(content="🔍 Đang kiểm tra tài khoản...", author="Kumiko")
         await loading_msg.send()
         
-        try:
-            # Check if user exists (run in executor to avoid blocking)
-            loop = asyncio.get_event_loop()
-            result = await loop.run_in_executor(None, check_user_exists, user_id)
-            
-            print(f"[DEBUG] Login check result: {result}")
-            
+        # Check if user exists
+        if check_user_exists(user_id):
             await loading_msg.remove()
-            
-            # Check if API call was successful
-            if not result.get("success", False):
-                await cl.Message(
-                    content=f"❌ Lỗi kiểm tra tài khoản: {result.get('error', 'Unknown error')}\n\n💡 API URL: `{API_BASE_URL}`",
-                    author="Kumiko"
-                ).send()
-                return
-            
-            # Check if user exists
-            if result.get("exists", False):
-                confirmed_user_id = result.get("user_id", user_id)
-                print(f"[DEBUG] Login successful for user: {confirmed_user_id}")
-                await show_chat_interface(confirmed_user_id)
-            else:
-                await cl.Message(
-                    content=f"❌ User ID `{user_id}` không tồn tại!\n\nVui lòng đăng ký bằng: `/register {user_id}`",
-                    author="Kumiko"
-                ).send()
-                
-        except Exception as e:
-            print(f"[ERROR] Exception in login: {type(e).__name__} - {e}")
+            await show_chat_interface(user_id)
+        else:
             await loading_msg.remove()
             await cl.Message(
-                content=f"❌ Lỗi khi kiểm tra tài khoản: {str(e)}\n\n💡 Tip: Kiểm tra API_BASE_URL và kết nối mạng\n\nAPI URL: `{API_BASE_URL}`",
+                content=f"❌ User ID `{user_id}` không tồn tại!\n\nVui lòng đăng ký bằng: `/register {user_id}`",
                 author="Kumiko"
             ).send()
     
@@ -351,10 +273,8 @@ async def handle_command(command: str):
         loading_msg = cl.Message(content="📝 Đang đăng ký tài khoản...", author="Kumiko")
         await loading_msg.send()
         
-        # Register user (run in executor to avoid blocking)
-        loop = asyncio.get_event_loop()
-        result = await loop.run_in_executor(None, register_user, user_id)
-        
+        # Register user
+        result = register_user(user_id)
         await loading_msg.remove()
         
         if result["success"]:
@@ -399,44 +319,43 @@ Hệ thống sẽ trả lời và cung cấp tài liệu tham khảo.
 
 async def process_chat_message(query: str, user_id: str):
     """Process chat message and display response"""
-    # Create empty message to show loading state
-    msg = cl.Message(content="")
-    await msg.send()
+    # Show thinking message
+    thinking_msg = cl.Message(content="🤔 Đang suy nghĩ...", author="Kumiko")
+    await thinking_msg.send()
     
-    try:
-        # Send request to API with async support
-        result = await send_chat_message_async(query, user_id)
-        
-        if not result["success"]:
-            msg.content = f"❌ Lỗi: {result['message']}"
-            await msg.update()
-            return
-        
-        answer = result["answer"]
-        references = result["references"]
-        
-        # Store references in session for the button
-        cl.user_session.set("last_references", references)
-        
-        # Create action button
-        actions = [
-            cl.Action(
-                name="show_references",
-                value="show",
-                label="𝄢 Tài liệu tham khảo",
-                description="Xem các nguồn tham khảo",
-                payload={"references": references}
-            )
-        ]
-        
-        # Update message with answer and actions
-        msg.content = answer
-        msg.actions = actions
-        await msg.update()
-        
-    except Exception as e:
-        msg.content = f"❌ Lỗi không mong muốn: {str(e)}\n\nVui lòng thử lại!"
-        await msg.update()
+    # Send request to API
+    result = send_chat_message(query, user_id)
+    
+    # Remove thinking message
+    await thinking_msg.remove()
+    
+    if not result["success"]:
+        await cl.Message(content=f"❌ Lỗi: {result['message']}", author="Kumiko").send()
+        return
+    
+    answer = result["answer"]
+    references = result["references"]
+    
+    # Store references in session for the button
+    cl.user_session.set("last_references", references)
+    
+    # Create message with action button
+    actions = [
+        cl.Action(
+            name="show_references",
+            value="show",
+            label="𝄢 Tài liệu tham khảo",
+            description="Xem các nguồn tham khảo",
+            payload={"references": references}
+        )
+    ]
+    
+    # Send answer with reference button
+    await cl.Message(
+        content=answer,
+        actions=actions,
+        author="Kumiko"
+    ).send()
 
 
 @cl.action_callback("show_references")