Update app.py

app.py

@@ -1,21 +1,22 @@
 import streamlit as st
-from langchain_community.llms import HuggingFaceHub  # Use updated import
+from langchain_community.llms import HuggingFaceHub  # Updated import
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 import os
 from datetime import datetime, timedelta
 from dotenv import load_dotenv
 
+# Print version information for debugging
+try:
+    import langchain_community
+
+    print(f"langchain_community version: {langchain_community.__version__}")
+except Exception as e:
+    print(f"Error getting version info: {e}")
+
 # 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",
@@ -50,21 +51,15 @@ language = st.selectbox(
     key="lang_select"
 )
 
-# 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 found in environment variables. Please:
-        1. Create a .env file with HUGGINGFACEHUB_API_TOKEN=your_token_here
-        2. Set the environment variable using setx on Windows
-        3. On Hugging Face Spaces, add it as a secret
-    """)
-    
-    # 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:
+# Get token from environment
+token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
+
+# For development, allow manual token entry if not found in environment
+if not token:
+    st.warning("API token not found in environment variables.")
+    token = st.text_input("Enter your Hugging Face API token:", type="password")
+    if not token:
+        st.error("Token is required to proceed.")
         st.stop()
 
 # Enhanced model configuration
@@ -73,28 +68,23 @@ model_config = {
         "repo_id": "google/flan-t5-base",  # Using a smaller model for testing
         "params": {
             "temperature": 0.7,
-            "max_length": 512,
-            "max_new_tokens": 300,
-            "repetition_penalty": 1.2
+            "max_length": 512
         }
     },
     "العربية": {
         "repo_id": "aubmindlab/aragpt2-medium",  # Using a smaller model for testing
         "params": {
             "temperature": 0.6,
-            "max_length": 1024,
-            "max_new_tokens": 400,
-            "repetition_penalty": 1.3
+            "max_length": 1024
         }
     }
 }
 
 # Initialize the language model with enhanced error handling
 try:
-    # This is the key fix - correct parameter name is huggingfacehub_api_token
+    # Simple initialization - let the library handle the token from env vars
     llm = HuggingFaceHub(
         repo_id=model_config[language]["repo_id"],
-        huggingfacehub_api_token=hf_token,  # CORRECT PARAMETER NAME
         model_kwargs=model_config[language]["params"]
     )
 
@@ -103,13 +93,20 @@ try:
         memory=st.session_state.memory,
         verbose=False
     )
-    
-    # Show success message
-    st.success("Successfully connected to Hugging Face model!")
-    
+
+    st.success("Model connected successfully!")
+
 except Exception as e:
     error_msg = str(e)
-    st.error(f"Initialization error: {error_msg}")
+    st.error(f"Error initializing model: {error_msg}")
+
+    if "token" in error_msg.lower() or "api" in error_msg.lower():
+        st.info("""
+        Make sure the token is set correctly:
+        1. Create a Hugging Face account and get a token
+        2. Set it as an environment variable: `setx HUGGINGFACEHUB_API_TOKEN "your_token"`
+        3. Restart your terminal after setting the environment variable
+        """)
     st.stop()
 
 # Display chat history with improved formatting
@@ -214,4 +211,11 @@ with st.sidebar:
     if st.button("Reset Conversation" if language == "English" else "إعادة ضبط المحادثة"):
         st.session_state.messages = []
         st.session_state.memory.clear()
-        st.experimental_rerun()
+        st.experimental_rerun()
+
+    # Add a manual token entry option in sidebar for easy updating
+    st.subheader("API Configuration")
+    new_token = st.text_input("Update API Token:", type="password")
+    if st.button("Save Token"):
+        os.environ["HUGGINGFACEHUB_API_TOKEN"] = new_token
+        st.success("Token updated! Restart the app to apply.")