Update app.py

app.py

@@ -1,61 +1,59 @@
 import streamlit as st
+from transformers import pipeline
+import urduhack
 
-# 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": "خوش آمدید! کیا میں آپ کی مزید مدد کر سکتا ہوں؟"
-    }
-}
-
-# 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"
-
-# Initialize chat history
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
+# Set page configuration
+st.set_page_config(page_title="Urdu AI Chatbot", page_icon="🤖")
 
-# User 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("براہ کرم ایک پیغام لکھیں۔")
+# Title and description
+st.title("اردو AI چیٹ بوٹ")
+st.write("یہ ایک سادہ چیٹ بوٹ ہے جو اردو زبان میں جواب دیتا ہے۔ اپنا سوال درج کریں اور جواب حاصل کریں!")
+
+# Load the Urdu model from Hugging Face
+@st.cache_resource
+def load_model():
+    # Using urduhack/roberta-urdu-small for fill-mask task as an example
+    model = pipeline("fill-mask", model="urduhack/roberta-urdu-small", tokenizer="urduhack/roberta-urdu-small")
+    return model
+
+model = load_model()
+
+# Function to process input and generate response
+def generate_response(user_input):
+    try:
+        # Replace [MASK] in the input for the fill-mask task
+        if "[MASK]" not in user_input:
+            user_input = user_input + " [MASK]"
+        result = model(user_input)
+        # Return the top prediction
+        return result[0]["sequence"]
+    except Exception as e:
+        return f"معذرت، کچھ غلطی ہوئی: {str(e)}"
+
+# Chat interface
+if "messages" not in st.session_state:
+    st.session_state.messages = [{"role": "assistant", "content": "سلام! میں آپ کی مدد کیسے کر سکتا ہوں؟"}]
 
 # Display chat history
-st.write("چیٹ کی تاریخ:")
-for message in st.session_state.chat_history:
-    st.write(message)
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.write(message["content"])
+
+# User input
+if prompt := st.chat_input("اپنا سوال یہاں لکھیں..."):
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.write(prompt)
+
+    # Generate and display assistant response
+    with st.chat_message("assistant"):
+        with st.spinner("غور کر رہا ہوں..."):
+            response = generate_response(prompt)
+        st.write(response)
+        st.session_state.messages.append({"role": "assistant", "content": response})
+
+# Sidebar with additional info
+st.sidebar.title("معلومات")
+st.sidebar.write("یہ چیٹ بوٹ Hugging Face کے `urduhack/roberta-urdu-small` ماڈل کا استعمال کرتا ہے۔")
+st.sidebar.write("مزید ماڈلز کے لیے [Hugging Face](https://huggingface.co/models) دیکھیں۔")