Update app.py

app.py

@@ -1,52 +1,61 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Set page config for Urdu (right-to-left support)
-st.set_page_config(page_title="اردو AI", layout="centered")
-
-# Load the Urdu model and tokenizer
-model_name = "m3hrdadfi/mt5-small-finetuned-urdu-sentiment"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
-# Custom CSS for Urdu font and RTL support
-st.markdown("""
-<style>
-    body {
-        font-family: "Noto Sans Urdu", sans-serif;
-        direction: rtl;
-        text-align: right;
+# Set the title of the app
+st.title("Urdu AI Chatbot 🤖")
+
+# Define intents and responses (keyword-based)
+intents = {
+    "greeting": {
+        "keywords": ["ہیلو", "سلام", "آداب", "حال", "کیسے ہو", "کیسے ہیں"],
+        "response": "السلام علیکم! میں ٹھیک ہوں، شکریہ۔ آپ کیسے ہیں؟"
+    },
+    "farewell": {
+        "keywords": ["الوداع", "خدا حافظ", "بعد میں ملتے ہیں"],
+        "response": "خدا حافظ! آپ کا دن اچھا گزرے۔"
+    },
+    "thanks": {
+        "keywords": ["شکریہ", "بہت شکریہ", "مہربانی"],
+        "response": "خوش آمدید! کیا میں آپ کی مزید مدد کر سکتا ہوں؟"
     }
-</style>
-""", unsafe_allow_html=True)
+}
 
-# Title
-st.title("🕌 اردو AI مددگار")
+# Function to detect intent using keywords
+def detect_intent(user_input):
+    user_input = user_input.lower()
+    for intent, data in intents.items():
+        for keyword in data["keywords"]:
+            if keyword in user_input:
+                return intent
+    return "unknown"
 
-# Chat history
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Display chat history
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"], unsafe_allow_html=True)
+# Initialize chat history
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
 
 # User input
-user_input = st.chat_input("اپنا پیغام یہاں لکھیں...")
+user_input = st.text_input("آپ کا پیغام یہاں لکھیں:", "ہیلو، آپ کیسے ہیں؟")
+
+# Send button
+if st.button("بھیجیں"):
+    if user_input:
+        # Add user message to chat history
+        st.session_state.chat_history.append(f"آپ: {user_input}")
+        
+        # Detect intent using keywords (no model needed)
+        intent = detect_intent(user_input)
+        
+        # Get response based on intent
+        if intent != "unknown":
+            response = intents[intent]["response"]
+        else:
+            response = "معذرت، میں اس سوال کا جواب نہیں دے سکتا۔ براہ کرم ایک آسان سوال پوچھیں۔"
+        
+        # Add AI response to chat history
+        st.session_state.chat_history.append(f"AI: {response}")
+    else:
+        st.warning("براہ کرم ایک پیغام لکھیں۔")
 
-# Generate response
-if user_input:
-    # Add user message to history
-    st.session_state.messages.append({"role": "user", "content": user_input})
-    
-    # Tokenize input and generate response
-    inputs = tokenizer.encode(user_input, return_tensors="pt")
-    outputs = model.generate(inputs, max_length=100)
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
-    # Add AI response to history
-    st.session_state.messages.append({"role": "assistant", "content": response})
-    
-    # Rerun to update the UI
-    st.rerun()
+# Display chat history
+st.write("چیٹ کی تاریخ:")
+for message in st.session_state.chat_history:
+    st.write(message)