Upload folder using huggingface_hub

app.py

@@ -1,37 +1,87 @@
-
 import streamlit as st
 import requests
 import pandas as pd
+import base64
+import logging
+import time
+
+# -------------------------------------------------------
+# 🧾 Logging Configuration
+# -------------------------------------------------------
+logging.basicConfig(
+    level=logging.INFO,
+    format="[%(asctime)s] [%(levelname)s] %(message)s",
+    datefmt="%Y-%m-%d %H:%M:%S",
+)
+logger = logging.getLogger("ChatBot-Frontend")
 
-# Backend URL (if deployed together on Hugging Face, use relative URL)
-BACKEND_URL = "https://gauravsahu1990-Chabot_Backend.hf.space/api/ask"  # Change to your Space/EC2 endpoint if remote
+logger.info("🚀 Starting ChatBot frontend...")
 
-st.set_page_config(page_title="ChatSQL Assistant", page_icon="🧠", layout="wide")
+# -------------------------------------------------------
+# ⚙️ Backend URL (update for your Hugging Face Space)
+# -------------------------------------------------------
+BACKEND_URL = "https://gauravsahu1990-Chabot_Backend.hf.space/api/ask"  # Adjust if hosted elsewhere
 
-st.title("🧠 ChatSQL Assistant")
-st.write("Ask your question about the database — I’ll show the result and chart!")
+# -------------------------------------------------------
+# 🧠 Streamlit App Setup
+# -------------------------------------------------------
+st.set_page_config(page_title="Chat Assistant", page_icon="🧠", layout="wide")
+st.title("🧠 Chat Assistant")
+st.write("Ask your question — I’ll show the result and chart!")
 
-# User input
+# -------------------------------------------------------
+# 💬 User Input
+# -------------------------------------------------------
 question = st.text_input("💬 Ask your question:", placeholder="e.g., Show total order amount by customer")
 
+# -------------------------------------------------------
+# 🎛️ Query Submission
+# -------------------------------------------------------
 if st.button("Ask"):
     if not question.strip():
         st.warning("Please enter a question.")
     else:
         with st.spinner("Thinking..."):
+            start_time = time.time()
             try:
-                response = requests.post(BACKEND_URL, json={"question": question})
-                data = response.json()
+                logger.info(f"🗨️ Sending question to backend: {question}")
+                response = requests.post(BACKEND_URL, json={"question": question}, timeout=60)
+                elapsed = round(time.time() - start_time, 2)
+                logger.info(f"⏱️ Backend responded in {elapsed}s, status {response.status_code}")
 
-                st.markdown("### 📊 Result")
-                st.markdown(data.get("answer", ""), unsafe_allow_html=True)
-
-                if data.get("chart"):
-                    st.markdown("### 📈 Visualization")
-                    img_bytes = base64.b64decode(data["chart"])
-                    st.image(img_bytes, use_container_width=True)
+                if response.status_code != 200:
+                    st.error(f"❌ Backend returned HTTP {response.status_code}")
+                    logger.error(f"❌ Backend response: {response.text}")
                 else:
-                    st.info("No chart available for this query.")
+                    try:
+                        data = response.json()
+                        logger.info(f"✅ JSON parsed successfully: keys={list(data.keys())}")
+
+                        st.markdown("### 📊 Result")
+                        st.markdown(data.get("answer", ""), unsafe_allow_html=True)
+
+                        if data.get("chart"):
+                            st.markdown("### 📈 Visualization")
+                            img_bytes = base64.b64decode(data["chart"])
+                            st.image(img_bytes, use_container_width=True)
+                            logger.info("📈 Chart displayed successfully.")
+                        else:
+                            st.info("No chart available for this query.")
+                            logger.info("ℹ️ No chart returned from backend.")
+
+                    except Exception as json_err:
+                        st.error("⚠️ Failed to parse backend response. Check logs for details.")
+                        logger.exception(f"❌ JSON parsing error: {json_err}")
+                        logger.error(f"Raw response: {response.text}")
+
+            except requests.exceptions.ConnectionError as conn_err:
+                st.error("⚠️ Could not connect to backend. Is it running?")
+                logger.exception(f"❌ Connection error: {conn_err}")
+
+            except requests.exceptions.Timeout:
+                st.error("⚠️ Backend took too long to respond.")
+                logger.error("⏰ Request timeout after 60s.")
 
             except Exception as e:
-                st.error(f"⚠️ Error: {e}")
+                st.error(f"⚠️ Unexpected error: {e}")
+                logger.exception(f"❌ Unexpected frontend error: {e}")