Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from langchain.llms import HuggingFaceHub
+from langchain_community.llms import HuggingFaceHub  # Use updated import
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 import os
@@ -9,6 +9,13 @@ from dotenv import load_dotenv
 # Load environment variables - MUST be at the top
 load_dotenv('.env')  # Explicitly load from .env file
 
+# Debugging - check if token is available (remove in production)
+token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
+if token:
+    print("Token found! Length:", len(token))
+else:
+    print("WARNING: Token not found in environment variables!")
+
 # Configure page
 st.set_page_config(
     page_title="Tourism Chatbot",
@@ -43,20 +50,27 @@ language = st.selectbox(
     key="lang_select"
 )
 
-# Hugging Face API token configuration - PRODUCTION READY
-HF_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
-if not HF_TOKEN:
+# Get token - try both environment variable and direct input
+hf_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
+
+# Display token status
+if not hf_token:
     st.error("""
-        API token not configured properly. Please:
+        API token not found in environment variables. Please:
         1. Create a .env file with HUGGINGFACEHUB_API_TOKEN=your_token_here
-        2. On Hugging Face Spaces, add it as a secret named HUGGINGFACEHUB_API_TOKEN
+        2. Set the environment variable using setx on Windows
+        3. On Hugging Face Spaces, add it as a secret
     """)
-    st.stop()
+    
+    # Allow token input directly in the app (for development only)
+    hf_token = st.text_input("Enter your Hugging Face token:", type="password")
+    if not hf_token:
+        st.stop()
 
 # Enhanced model configuration
 model_config = {
     "English": {
-        "repo_id": "google/flan-t5-xxl",
+        "repo_id": "google/flan-t5-base",  # Using a smaller model for testing
         "params": {
             "temperature": 0.7,
             "max_length": 512,
@@ -65,7 +79,7 @@ model_config = {
         }
     },
     "العربية": {
-        "repo_id": "aubmindlab/aragpt2-mega",
+        "repo_id": "aubmindlab/aragpt2-medium",  # Using a smaller model for testing
         "params": {
             "temperature": 0.6,
             "max_length": 1024,
@@ -77,10 +91,10 @@ model_config = {
 
 # Initialize the language model with enhanced error handling
 try:
-    # This is the corrected line - using the proper parameter name
+    # This is the key fix - correct parameter name is huggingfacehub_api_token
     llm = HuggingFaceHub(
         repo_id=model_config[language]["repo_id"],
-        huggingfacehub_api_token=HF_TOKEN,
+        huggingfacehub_api_token=hf_token,  # CORRECT PARAMETER NAME
         model_kwargs=model_config[language]["params"]
     )
 
@@ -89,14 +103,13 @@ try:
         memory=st.session_state.memory,
         verbose=False
     )
+    
+    # Show success message
+    st.success("Successfully connected to Hugging Face model!")
+    
 except Exception as e:
     error_msg = str(e)
-    if "API token" in error_msg:
-        st.error("Invalid API token. Please check your Hugging Face token.")
-    elif "repo_id" in error_msg:
-        st.error("Model loading failed. The specified model may not be available.")
-    else:
-        st.error(f"Initialization error: {error_msg}")
+    st.error(f"Initialization error: {error_msg}")
     st.stop()
 
 # Display chat history with improved formatting