Update app.py

app.py

@@ -10,12 +10,6 @@ LLM_API = os.environ.get("LLM_API")
 LLM_URL = os.environ.get("LLM_URL")
 USER_ID = "HuggingFace Space"
 
-# 設置重試次數和延遲
-MAX_RETRIES = 3
-RETRY_DELAY = 2
-TIMEOUT_DURATION = 180  # 超時時間設定為 180 秒
-MAX_RESPONSES = 15
-
 async def send_chat_message(LLM_URL, LLM_API, user_input):
     payload = {
         "inputs": {},
@@ -26,50 +20,42 @@ async def send_chat_message(LLM_URL, LLM_API, user_input):
     }
     print("Sending chat message payload:", payload)
 
-    for attempt in range(MAX_RETRIES):
-        async with aiohttp.ClientSession() as session:
-            try:
-                async with session.post(
-                    url=f"{LLM_URL}/chat-messages",
-                    headers={"Authorization": f"Bearer {LLM_API}"},
-                    json=payload,
-                    timeout=aiohttp.ClientTimeout(total=TIMEOUT_DURATION)
-                ) as response:
-                    if response.status != 200:
-                        print(f"Error: {response.status}")
-                        continue
+    async with aiohttp.ClientSession() as session:
+        try:
+            async with session.post(
+                url=f"{LLM_URL}/chat-messages",
+                headers={"Authorization": f"Bearer {LLM_API}"},
+                json=payload,
+                timeout=aiohttp.ClientTimeout(total=180)
+            ) as response:
+                if response.status != 200:
+                    print(f"Error: {response.status}")
+                    return f"Error: Status code {response.status}"
 
-                    full_response = []
-                    async for line in response.content.iter_chunked(2048):
-                        line = line.decode('utf-8').strip()
-                        if not line or "data: " not in line:
-                            continue
-                        try:
-                            print("Received line:", line)
-                            json_str = line.split("data: ", 1)[-1]
-                            data = json.loads(json_str)
-                            if "answer" in data:
-                                decoded_answer = urllib.parse.unquote(data["answer"])
-                                full_response.append(decoded_answer)
-                                if len(full_response) >= MAX_RESPONSES:
-                                    break
-                        except (IndexError, json.JSONDecodeError) as e:
-                            print(f"Error parsing line: {line}, error: {e}")
-                            continue
+                full_response = []
+                async for line in response.content.iter_chunked(2048):
+                    line = line.decode('utf-8').strip()
+                    if not line or "data: " not in line:
+                        continue
+                    try:
+                        data = json.loads(line.split("data: ")[1])
+                        if "answer" in data:
+                            decoded_answer = urllib.parse.unquote(data["answer"])
+                            full_response.append(decoded_answer)
+                    except (IndexError, json.JSONDecodeError) as e:
+                        print(f"Error parsing line: {line}, error: {e}")
+                        continue
 
-                    if full_response:
-                        return ''.join(full_response).strip()
-                    else:
-                        return "Error: No response found in the response"
-            except asyncio.TimeoutError:
-                print(f"Attempt {attempt + 1} timed out. Retrying...")
-                await asyncio.sleep(RETRY_DELAY)
-                continue
-            except Exception as e:
-                print("Exception occurred in send_chat_message:")
-                print(traceback.format_exc())
-                return f"Exception: {e}"
-    return "Error: Reached maximum retry attempts."
+                if full_response:
+                    return ''.join(full_response).strip()
+                else:
+                    return "Error: No response found in the response"
+        except aiohttp.ClientConnectorError:
+            return "Error: Cannot connect to the API server. Please check the URL and server status."
+        except Exception as e:
+            print("Exception occurred in send_chat_message:")
+            print(traceback.format_exc())
+            return f"Exception: {e}"
 
 async def handle_input(user_input):
     print(f"Handling input: {user_input}")
@@ -98,7 +84,7 @@ examples = [
 
 
 
-TITLE = """<h1>Social Media Trends 💬 分析社群相關資訊，並判斷其正、負、中立等評價及趨勢 </h1>"""
+TITLE = """<h1>Social Media Trends 💬 分析社群相關資訊，並判斷其正、負、中立等評價及趨勢 (數據大會跑很久)) </h1>"""
 SUBTITLE = """<h2><a href='https://www.twman.org' target='_blank'>TonTon Huang Ph.D. @ 2024/11 </a><br></h2>"""
 LINKS = """
 <a href='https://github.com/Deep-Learning-101' target='_blank'>Deep Learning 101 Github</a> | <a href='http://deeplearning101.twman.org' target='_blank'>Deep Learning 101</a> | <a href='https://www.facebook.com/groups/525579498272187/' target='_blank'>台灣人工智慧社團 FB</a> | <a href='https://www.youtube.com/c/DeepLearning101' target='_blank'>YouTube</a><br>