Update app.py

app.py

@@ -1,72 +1,61 @@
 import streamlit as st
-from transformers import DistilBertTokenizer, DistilBertForQuestionAnswering
-import torch
 
-# Set page configuration
-st.set_page_config(page_title="Urdu AI Chatbot", page_icon="🤖")
+# 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 = []
 
-# Title and description
-st.title("اردو AI چیٹ بوٹ")
-st.write("یہ ایک سادہ چیٹ بوٹ ہے جو اردو زبان میں جواب دیتا ہے۔ اپنا سوال درج کریں!")
-
-# Load a lightweight multilingual model
-@st.cache_resource
-def load_model():
-    try:
-        tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-multilingual-cased")
-        model = DistilBertForQuestionAnswering.from_pretrained("distilbert-base-multilingual-cased")
-        return model, tokenizer
-    except Exception as e:
-        st.error(f"ماڈل لوڈ کرنے میں خرابی: {str(e)}")
-        return None, None
-
-model, tokenizer = load_model()
-if model is None or tokenizer is None:
-    st.error("ماڈل یا ٹوکنائزر لوڈ نہیں ہوا۔ براہ کرم لاگز چیک کریں۔")
-    st.stop()
+# User input
+user_input = st.text_input("آپ کا پیغام یہاں لکھیں:", "ہیلو، آپ کیسے ہیں؟")
 
-# Function to generate a simple response
-def generate_response(user_input):
-    try:
-        # Encode the input
-        inputs = tokenizer(user_input, return_tensors="pt", truncation=True, padding=True)
-        outputs = model(**inputs)
-        # Since this is a QA model, we'll simulate a response
-        start_scores = outputs.start_logits
-        end_scores = outputs.end_logits
-        start_idx = torch.argmax(start_scores)
-        end_idx = torch.argmax(end_scores) + 1
-        answer_tokens = inputs["input_ids"][0][start_idx:end_idx]
-        response = tokenizer.decode(answer_tokens, skip_special_tokens=True)
+# Send button
+if st.button("بھیجیں"):
+    if user_input:
+        # Add user message to chat history
+        st.session_state.chat_history.append(f"آپ: {user_input}")
         
-        # If the response is empty or too short, provide a fallback
-        if not response or len(response) < 3:
-            return "میں سمجھ گیا، لیکن براہ کرم مزید وضاحت کریں۔"
-        return response
-    except Exception as e:
-        return f"معذرت، کچھ غلطی ہوئی: {str(e)}"
-
-# Chat interface
-if "messages" not in st.session_state:
-    st.session_state.messages = [{"role": "assistant", "content": "سلام! میں آپ کی مدد کے لیے حاضر ہوں۔"}]
+        # 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("براہ کرم ایک پیغام لکھیں۔")
 
 # Display chat history
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.write(message["content"])
-
-# User input
-if prompt := st.chat_input("اپنا سوال یہاں لکھیں..."):
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.write(prompt)
-
-    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
-st.sidebar.title("معلومات")
-st.sidebar.write("یہ `distilbert-base-multilingual-cased` م��ڈل استعمال کرتا ہے۔")
+st.write("چیٹ کی تاریخ:")
+for message in st.session_state.chat_history:
+    st.write(message)