Update app.py

app.py

@@ -171,8 +171,7 @@ if INDEX_NAME not in existing_indexes:
 else:
 
     logger.info(
-        "Existing Pinecone index found. "
-        "NOT deleting it."
+        "Existing Pinecone index found. NOT deleting it."
     )
 
 
@@ -238,47 +237,39 @@ if vector_count == 0:
         }
     ).load_data()
 
-
     if not documents:
 
         raise ValueError(
             "No PDF documents found inside Data folder"
         )
 
-
     logger.info(
         f"Loaded {len(documents)} document sections"
     )
 
-
     storage_context = StorageContext.from_defaults(
         vector_store=vector_store
     )
 
-
     logger.info(
         "Creating embeddings and storing them in Pinecone..."
     )
 
-
     index = VectorStoreIndex.from_documents(
         documents,
         storage_context=storage_context,
     )
 
-
     logger.info(
         "DDS HR Handbook indexed successfully."
     )
 
-
 else:
 
     logger.info(
         "Using existing Pinecone vectors."
     )
 
-
     index = VectorStoreIndex.from_vector_store(
         vector_store=vector_store
     )
@@ -302,19 +293,15 @@ def query_doc(question):
     if not question:
         return "Please enter an HR question."
 
-
     question = question.strip()
 
-
     if not question:
         return "Please enter an HR question."
 
-
     logger.info(
         f"HR question received: {question}"
     )
 
-
     try:
 
         response = query_engine.query(
@@ -329,7 +316,6 @@ def query_doc(question):
 
         return answer
 
-
     except Exception as e:
 
         logger.exception(
@@ -343,7 +329,7 @@ def query_doc(question):
 
 
 # ============================================================
-# 12. TELEGRAM FUNCTIONS
+# 12. TELEGRAM URL
 # ============================================================
 
 def telegram_url(method):
@@ -356,37 +342,52 @@ def telegram_url(method):
 
 
 # ============================================================
-# 13. TEST TELEGRAM CONNECTION
+# 13. TEST TELEGRAM CONNECTION FROM HUGGING FACE
 # ============================================================
 
 def test_telegram_connection():
 
     if not TELEGRAM_BOT_TOKEN:
+
+        logger.error(
+            "TELEGRAM TEST: Token is missing."
+        )
+
         return False
 
+    logger.info(
+        "===================================="
+    )
+
+    logger.info(
+        "TESTING HUGGING FACE -> TELEGRAM"
+    )
+
+    logger.info(
+        f"TELEGRAM TOKEN FOUND: {bool(TELEGRAM_BOT_TOKEN)}"
+    )
 
     try:
 
         response = requests.get(
             telegram_url("getMe"),
-            timeout=10
+            timeout=30
+        )
+
+        logger.info(
+            f"TELEGRAM TEST STATUS: {response.status_code}"
         )
 
         response.raise_for_status()
 
         data = response.json()
 
-
         if data.get("ok"):
 
             bot = data["result"]
 
             logger.info(
-                "===================================="
-            )
-
-            logger.info(
-                "TELEGRAM CONNECTED"
+                "TELEGRAM TEST RESULT: SUCCESS"
             )
 
             logger.info(
@@ -403,18 +404,32 @@ def test_telegram_connection():
 
             return True
 
+        logger.error(
+            f"TELEGRAM TEST RESULT: API ERROR: {data}"
+        )
+
+        return False
+
+    except requests.exceptions.Timeout as e:
 
         logger.error(
-            f"Telegram rejected token: {data}"
+            f"TELEGRAM TEST FAILED: TIMEOUT: {repr(e)}"
         )
 
         return False
 
+    except requests.exceptions.RequestException as e:
+
+        logger.error(
+            f"TELEGRAM TEST FAILED: NETWORK ERROR: {repr(e)}"
+        )
+
+        return False
 
     except Exception as e:
 
         logger.exception(
-            f"Telegram connection error: {e}"
+            f"TELEGRAM TEST FAILED: {repr(e)}"
         )
 
         return False
@@ -424,55 +439,35 @@ def test_telegram_connection():
 # 14. SEND TELEGRAM MESSAGE
 # ============================================================
 
-def send_telegram_message(
-    chat_id,
-    text
-):
+def send_telegram_message(chat_id, text):
 
     if not TELEGRAM_BOT_TOKEN:
         return
 
-
     text = str(text)
 
-
-    # Telegram messages have a length limit,
-    # so split long answers.
+    # Telegram messages have a length limit.
+    # Split long answers into smaller pieces.
     chunks = [
-
         text[i:i + 4000]
-
-        for i in range(
-            0,
-            len(text),
-            4000
-        )
-
+        for i in range(0, len(text), 4000)
     ]
 
-
     for chunk in chunks:
 
         try:
 
             response = requests.post(
-
-                telegram_url(
-                    "sendMessage"
-                ),
-
+                telegram_url("sendMessage"),
                 json={
                     "chat_id": chat_id,
                     "text": chunk,
                 },
-
-                timeout=15
+                timeout=30
             )
 
-
             response.raise_for_status()
 
-
         except Exception as e:
 
             logger.exception(
@@ -481,7 +476,7 @@ def send_telegram_message(
 
 
 # ============================================================
-# 15. SHOW TYPING...
+# 15. SHOW TELEGRAM TYPING
 # ============================================================
 
 def telegram_typing(chat_id):
@@ -489,31 +484,21 @@ def telegram_typing(chat_id):
     try:
 
         requests.post(
-
-            telegram_url(
-                "sendChatAction"
-            ),
-
+            telegram_url("sendChatAction"),
             json={
                 "chat_id": chat_id,
                 "action": "typing",
             },
-
-            timeout=5
+            timeout=10
         )
 
-
     except Exception:
 
         pass
 
 
 # ============================================================
-# 16. TELEGRAM POLLING
-#
-# IMPORTANT:
-# We deliberately use SHORT POLLING for this demo.
-# This avoids the 55-second Hugging Face timeout you saw.
+# 16. TELEGRAM POLLING LOOP
 # ============================================================
 
 def telegram_polling_loop():
@@ -527,10 +512,9 @@ def telegram_polling_loop():
 
         return
 
-
-    # ----------------------------------------------
-    # Verify token first
-    # ----------------------------------------------
+    # --------------------------------------------------------
+    # STEP 1: TEST HUGGING FACE -> TELEGRAM
+    # --------------------------------------------------------
 
     if not test_telegram_connection():
 
@@ -538,35 +522,33 @@ def telegram_polling_loop():
             "Telegram connection failed."
         )
 
+        logger.error(
+            "Web chatbot will continue running normally."
+        )
+
         return
 
 
-    # ----------------------------------------------
-    # Delete old webhook
-    # getUpdates will not work with a webhook active.
-    # ----------------------------------------------
+    # --------------------------------------------------------
+    # STEP 2: DELETE ANY OLD WEBHOOK
+    # --------------------------------------------------------
 
     try:
 
-        requests.post(
-
-            telegram_url(
-                "deleteWebhook"
-            ),
-
+        response = requests.post(
+            telegram_url("deleteWebhook"),
             json={
                 "drop_pending_updates": True
             },
-
-            timeout=10
+            timeout=30
         )
 
+        response.raise_for_status()
 
         logger.info(
             "Old Telegram webhook cleared."
         )
 
-
     except Exception as e:
 
         logger.warning(
@@ -574,6 +556,10 @@ def telegram_polling_loop():
         )
 
 
+    # --------------------------------------------------------
+    # STEP 3: START LISTENING FOR TELEGRAM MESSAGES
+    # --------------------------------------------------------
+
     logger.info(
         "===================================="
     )
@@ -602,28 +588,18 @@ def telegram_polling_loop():
                 "timeout": 0
             }
 
-
             if offset is not None:
-
                 params["offset"] = offset
 
 
-            # SHORT POLLING
             response = requests.get(
-
-                telegram_url(
-                    "getUpdates"
-                ),
-
+                telegram_url("getUpdates"),
                 params=params,
-
-                timeout=10
+                timeout=30
             )
 
-
             response.raise_for_status()
 
-
             data = response.json()
 
 
@@ -647,18 +623,14 @@ def telegram_polling_loop():
             for update in updates:
 
                 offset = (
-                    update["update_id"]
-                    + 1
+                    update["update_id"] + 1
                 )
 
-
                 message = update.get(
                     "message"
                 )
 
-
                 if not message:
-
                     continue
 
 
@@ -667,12 +639,10 @@ def telegram_polling_loop():
                     {}
                 )
 
-
                 chat_id = chat.get(
                     "id"
                 )
 
-
                 text = (
                     message.get(
                         "text"
@@ -682,19 +652,15 @@ def telegram_polling_loop():
 
 
                 if not chat_id:
-
                     continue
 
 
                 if not text:
 
                     send_telegram_message(
-
                         chat_id,
-
                         "Please send me a text "
                         "question about DDS HR policies."
-
                     )
 
                     continue
@@ -705,102 +671,83 @@ def telegram_polling_loop():
                 )
 
 
-                # ==================================
+                # ============================================
                 # /start
-                # ==================================
+                # ============================================
 
-                if text.startswith(
-                    "/start"
-                ):
+                if text.startswith("/start"):
 
                     send_telegram_message(
-
                         chat_id,
-
                         "Welcome to AYesha — "
                         "DDS Enterprise HR Chatbot.\n\n"
                         "Ask me a question about the "
                         "latest DDS HR policies.\n\n"
                         "For example:\n"
                         "What is the annual leave policy?"
-
                     )
 
                     continue
 
 
-                # ==================================
+                # ============================================
                 # /help
-                # ==================================
+                # ============================================
 
-                if text.startswith(
-                    "/help"
-                ):
+                if text.startswith("/help"):
 
                     send_telegram_message(
-
                         chat_id,
-
                         "Ask me questions about DDS HR "
                         "policies.\n\n"
                         "Examples:\n"
                         "• What is the annual leave policy?\n"
                         "• What is the probation policy?\n"
                         "• How do I request leave?"
-
                     )
 
                     continue
 
 
-                # ==================================
+                # ============================================
                 # OTHER COMMAND
-                # ==================================
+                # ============================================
 
                 if text.startswith("/"):
 
                     send_telegram_message(
-
                         chat_id,
-
                         "Please type your HR question "
                         "as a normal message."
-
                     )
 
                     continue
 
 
-                # ==================================
+                # ============================================
                 # NORMAL HR QUESTION
-                # ==================================
+                # ============================================
 
                 telegram_typing(
                     chat_id
                 )
 
-
                 answer = query_doc(
                     text
                 )
 
-
                 send_telegram_message(
-
                     chat_id,
                     answer
-
                 )
 
 
-            # Prevent aggressive API calling
+            # Prevent aggressive Telegram API calls
             time.sleep(1)
 
 
         except requests.exceptions.Timeout:
 
-            # Network timeout is not fatal.
-            # Simply try again.
             logger.warning(
                 "Telegram temporary timeout. Retrying..."
             )
@@ -840,21 +787,14 @@ def start_telegram_bot():
 
         return
 
-
     thread = threading.Thread(
-
         target=telegram_polling_loop,
-
         daemon=True,
-
         name="DDS-Telegram-Bot"
-
     )
 
-
     thread.start()
 
-
     logger.info(
         "Telegram background thread started."
     )
@@ -876,13 +816,10 @@ demo = gr.Interface(
     fn=query_doc,
 
     inputs=gr.Textbox(
-
         label="Ask a question about DDS HR policies",
-
         placeholder=(
             "Example: What is the annual leave policy?"
         )
-
     ),
 
     outputs=gr.Textbox(
@@ -906,9 +843,6 @@ demo = gr.Interface(
 if __name__ == "__main__":
 
     demo.launch(
-
         server_name="0.0.0.0",
-
         server_port=7860
-
     )