Update app.py

app.py

@@ -3,89 +3,168 @@ from langchain.llms import HuggingFaceHub
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 import os
+from datetime import datetime, timedelta
 
 # Configure page
-st.set_page_config(page_title="Tourism Chatbot", page_icon="🌍")
+st.set_page_config(
+    page_title="Tourism Chatbot", 
+    page_icon="🌍",
+    layout="wide"
+)
 
-# Title
-st.title("Tourism Assistant - مساعد السياحة")
+# Title with better styling
+st.markdown("""
+    <h1 style='text-align: center; color: #2E86C1;'>
+        Tourism Assistant - مساعد السياحة
+    </h1>
+""", unsafe_allow_html=True)
 
-# Initialize session state for memory
+# Initialize session states
 if "memory" not in st.session_state:
-    st.session_state.memory = ConversationBufferMemory(return_messages=True)
+    st.session_state.memory = ConversationBufferMemory(
+        return_messages=True,
+        memory_key="chat_history"
+    )
 
-# Initialize session state for chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Language selector
-language = st.selectbox("Choose Language / اختر اللغة", ["English", "العربية"])
-
-# Hugging Face API token
-hf_api_token = os.environ.get("HF_API_TOKEN", "")
-if not hf_api_token:
-    st.error("Hugging Face API token not found. Please set it in the Space secrets.")
-    st.stop()
-
-# Select the appropriate model based on language
-model_name = "bigscience/bloom" if language == "English" else "bigscience/bloom"
+if "last_request" not in st.session_state:
+    st.session_state.last_request = datetime.now() - timedelta(seconds=10)
 
-# Initialize the language model
-llm = HuggingFaceHub(
-    repo_id=model_name,
-    huggingface_api_token=hf_api_token,
-    model_kwargs={"temperature": 0.7, "max_length": 512}
+# Language selector with better styling
+language = st.selectbox(
+    "Choose Language / اختر اللغة", 
+    ["English", "العربية"],
+    key="lang_select"
 )
 
-# Initialize conversation chain
-conversation = ConversationChain(
-    llm=llm,
-    memory=st.session_state.memory,
-    verbose=True
-)
+# Hugging Face API token with better error handling
+try:
+    hf_api_token = os.environ["HF_API_TOKEN"]
+except KeyError:
+    st.error("""
+        Hugging Face API token not found. 
+        Please set it in the Space secrets as HF_API_TOKEN.
+    """)
+    st.stop()
+
+# Improved model selection with better Arabic support
+model_config = {
+    "English": {
+        "repo_id": "google/flan-t5-large",
+        "params": {"temperature": 0.7, "max_length": 512}
+    },
+    "العربية": {
+        "repo_id": "aubmindlab/aragpt2-base",
+        "params": {"temperature": 0.5, "max_length": 1024}
+    }
+}
+
+# Initialize the language model with error handling
+try:
+    llm = HuggingFaceHub(
+        repo_id=model_config[language]["repo_id"],
+        huggingfacehub_api_token=hf_api_token,
+        model_kwargs=model_config[language]["params"]
+    )
+    
+    conversation = ConversationChain(
+        llm=llm,
+        memory=st.session_state.memory,
+        verbose=False
+    )
+except Exception as e:
+    st.error(f"Failed to initialize model: {str(e)}")
+    st.stop()
 
-# Display chat history
+# Display chat history with avatars
 for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
+    avatar = "🧑" if message["role"] == "user" else "🤖"
+    with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
-# User input
-if language == "English":
-    prompt = st.chat_input("Ask about tourist destinations, recommendations, or travel tips...")
-else:
-    prompt = st.chat_input("اسأل عن الوجهات السياحية أو التوصيات أو نصائح السفر...")
+# Rate limiting check
+if (datetime.now() - st.session_state.last_request).seconds < 2:
+    st.warning("Please wait a moment before sending another message.")
+    st.stop()
+
+# User input with language-specific placeholders
+prompt = st.chat_input(
+    "Ask about tourist destinations..." if language == "English" 
+    else "اسأل عن الوجهات السياحية..."
+)
 
-# Process the user input
 if prompt:
+    # Update last request time
+    st.session_state.last_request = datetime.now()
+    
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
-
+    
     # Display user message
-    with st.chat_message("user"):
+    with st.chat_message("user", avatar="🧑"):
         st.markdown(prompt)
-
-    # Display assistant response with a spinner
-    with st.chat_message("assistant"):
-        with st.spinner("Thinking..."):
-            # Add context for tourism assistant
-            if language == "English":
-                full_prompt = f"You are a helpful tourism assistant. Answer the following question about travel and tourism: {prompt}"
-            else:
-                full_prompt = f"أنت مساعد سياحي مفيد. أجب عن السؤال التالي حول السفر والسياحة: {prompt}"
-
-            response = conversation.predict(input=full_prompt)
-            st.markdown(response)
-
-    # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
-
-# Sidebar with information
+    
+    # Display assistant response
+    with st.chat_message("assistant", avatar="🤖"):
+        with st.spinner("Generating response..." if language == "English" else "جارٍ تحضير الرد..."):
+            try:
+                # Enhanced prompt engineering
+                if language == "English":
+                    full_prompt = f"""You are an expert tourism assistant. Provide detailed, helpful information about:
+                    - Travel destinations
+                    - Cultural experiences
+                    - Local customs
+                    - Safety advice
+                    
+                    Question: {prompt}
+                    Answer:"""
+                else:
+                    full_prompt = f"""أنت مساعد سياحي خبير. قدم معلومات مفيدة ومفصلة حول:
+                    - الوجهات السياحية
+                    - التجارب الثقافية
+                    - العادات المحلية
+                    - نصائح السلامة
+                    
+                    السؤال: {prompt}
+                    الجواب:"""
+                
+                response = conversation.predict(input=full_prompt)
+                st.markdown(response)
+                
+                # Add to chat history
+                st.session_state.messages.append({"role": "assistant", "content": response})
+                
+            except Exception as e:
+                st.error(f"Error generating response: {str(e)}")
+                st.session_state.messages.append({
+                    "role": "assistant", 
+                    "content": "Sorry, I encountered an error. Please try again."
+                })
+
+# Enhanced sidebar with clear sections
 with st.sidebar:
-    st.header("About this Chatbot")
+    st.header("ℹ️ About this Chatbot")
+    
     if language == "English":
-        st.write(
-            "This is a tourism assistant that can help you with travel recommendations, destination information, and travel tips.")
-        st.write("Feel free to ask questions in English or Arabic!")
+        st.markdown("""
+            ### Features:
+            - Bilingual English/Arabic support
+            - Tourism expert knowledge
+            - Conversation memory
+            
+            **Note:** Responses may take 10-20 seconds to generate.
+        """)
     else:
-        st.write("هذا مساعد سياحي يمكنه مساعدتك في توصيات السفر ومعلومات الوجهة ونصائح السفر.")
-        st.write("لا تتردد في طرح الأسئلة باللغة الإنجليزية أو العربية!")
+        st.markdown("""
+            ### الميزات:
+            - دعم ثنائي اللغة الإنجليزية/العربية
+            - معرفة خبيرة بالسياحة
+            - ذاكرة المحادثة
+            
+            **ملاحظة:** قد يستغرق الرد 10-20 ثانية.
+        """)
+    
+    st.markdown("---")
+    st.markdown("Made with ❤️ using [Streamlit](https://streamlit.io/) and [LangChain](https://langchain.com/)")