Update app.py

app.py

@@ -1,45 +1,68 @@
 import streamlit as st
 
+# Initialize session state for dynamic responses
+if 'urdu_responses' not in st.session_state:
+    st.session_state.urdu_responses = {
+        "سلام": "وعلیکم السلام! آپ کیسے ہیں؟",
+        "السلام علیکم": "وعلیکم السلام! آپ کیسے ہیں؟",
+        "اسلام علیکم": "وعلیکم السلام! آپ کیسے ہیں؟",
+        "آپ کا نام کیا ہے": "میں ایک سادہ چیٹ بوٹ ہوں۔",
+        "پاکستان کا دارالحکومت کونسا شہر ہے": "پاکستان کا دارالحکومت اسلام آباد ہے۔",
+        "آپ کیسے ہیں": "میں ٹھیک ہوں، شکریہ! آپ کیسے ہیں؟",
+        "الوداع": "الوداع! آپ کا دن اچھا گزرے۔",
+        "ٹرمپ کے کتنے بچے ہیں": "ڈونلڈ ٹرمپ کے پانچ بچے ہیں: ڈونلڈ جونیئر، ایوانکا، ایرک، ٹفنی، اور بیرن۔",
+        "ٹرمپ کا پہلا نام کیا ہے": "ٹرمپ کا پہلا نام ڈونلڈ ہے۔",
+        "ٹرمپ کی بیوی کا نام کیا ہے": "ڈونلڈ ٹرمپ کی موجودہ بیوی کا نام میلانیا ٹرمپ ہے۔",
+        "ٹرمپ کب صدر بنے": "ڈونلڈ ٹرمپ 2017 میں امریکہ کے صدر بنے۔",
+        "ٹرمپ کا تعلق کس سیاسی جماعت سے ہے": "ڈونلڈ ٹرمپ کا تعلق ریپبلکن پارٹی سے ہے۔",
+        "امریکہ کا دارالحکومت کونسا شہر ہے": "امریکہ کا دارالحکومت واشنگٹن ڈی سی ہے۔",
+        "چاند پر پہلا انسان کب گیا": "چاند پر پہلا انسان 20 جولائی 1969 کو گیا تھا۔",
+        "دنیا کا سب سے بڑا سمندر کونسا ہے": "دنیا کا سب سے بڑا سمندر بحرالکاہل ہے۔",
+        "دنیا کا سب سے بڑا دریا کونسا ہے": "دنیا کا سب سے بڑا دریا دریائے ایمیزون ہے۔",
+        "دنیا کا سب سے بڑا ملک کونسا ہے": "دنیا کا سب سے بڑا ملک روس ہے۔",
+        "کمپیوٹر کس نے ایجاد کیا": "کمپیوٹر کی ایجاد کا سہرا چارلس بیبیج کو جاتا ہے۔",
+        "پانی کا کیمیائی فارمولا کیا ہے": "پانی کا کیمیائی فارمولا H₂O ہے۔",
+        "پاکستان کا قائد اعظم کون ہے": "پاکستان کے قائد اعظم محمد علی جناح ہیں۔"
+    }
+
 # Streamlit app title
-st.title("اردو زبان میں سادہ چیٹ بوٹ")
-
-# Define a dictionary of Urdu questions and answers
-urdu_responses = {
-    "سلام": "وعلیکم السلام! آپ کیسے ہیں؟",
-    "آپ کا نام کیا ہے؟": "میں ایک سادہ چیٹ بوٹ ہوں۔",
-    "پاکستان کا دارالحکومت کونسا شہر ہے؟": "پاکستان کا دارالحکومت اسلام آباد ہے۔",
-    "آپ کیسے ہیں؟": "میں ٹھیک ہوں، شکریہ! آپ کیسے ہیں؟",
-    "الوداع": "الوداع! آپ کا دن اچھا گزرے۔",
-    "ٹرمپ کے کتنے بچے ہیں؟": "ڈونلڈ ٹرمپ کے پانچ بچے ہیں: ڈونلڈ جونیئر، ایوانکا، ایرک، ٹفنی، اور بیرن۔",
-    "ٹرمپ کا پہلا نام کیا ہے؟": "ٹرمپ کا پہلا نام ڈونلڈ ہے۔",
-    "ٹرمپ کی بیوی کا نام کیا ہے؟": "ڈونلڈ ٹرمپ کی موجودہ بیوی کا نام میلانیا ٹرمپ ہے۔",
-    "ٹرمپ کب صدر بنے؟": "ڈونلڈ ٹرمپ 2017 میں امریکہ کے صدر بنے۔",
-    "ٹرمپ کا تعلق کس سیاسی جماعت سے ہے؟": "ڈونلڈ ٹرمپ کا تعلق ریپبلکن پارٹی سے ہے۔",
-    "امریکہ کا دارالحکومت کونسا شہر ہے؟": "امریکہ کا دارالحکومت واشنگٹن ڈی سی ہے۔",
-    "چاند پر پہلا انسان کب گیا؟": "چاند پر پہلا انسان 20 جولائی 1969 کو گیا تھا۔",
-    "دنیا کا سب سے بڑا سمندر کونسا ہے؟": "دنیا کا سب سے بڑا سمندر بحرالکاہل ہے۔",
-    "دنیا کا سب سے بڑا دریا کونسا ہے؟": "دنیا کا سب سے بڑا دریا دریائے ایمیزون ہے۔",
-    "دنیا کا سب سے بڑا ملک کونسا ہے؟": "دنیا کا سب سے بڑا ملک روس ہے۔",
-}
-
-# Input box for user to ask questions in Urdu
+st.title("🕌 اردو زبان میں سادہ چیٹ بوٹ")
+
+# Main chat interface
 user_input = st.text_input("اپنا سوال یہاں درج کریں:", "")
 
-# Function to get a response based on user input
 def get_response(user_input):
-    # Convert input to lowercase for case-insensitive matching
-    user_input = user_input.strip().lower()
+    # Process input
+    processed_input = user_input.strip().lower().replace('؟', '').replace('?', '')
     
-    # Check if the input matches any predefined question
-    for question, response in urdu_responses.items():
-        if user_input == question.lower():
+    # Check for matching questions
+    for question, response in st.session_state.urdu_responses.items():
+        if processed_input == question.lower().replace('؟', ''):
             return response
     
-    # Default response if no match is found
     return "معاف کیجئے، میں آپ کے سوال کا جواب نہیں دے سکتا۔"
 
-# Display AI response
+# Display conversation
 if user_input:
     st.write("**آپ کا سوال:**", user_input)
     bot_response = get_response(user_input)
-    st.write("**بوٹ کا جواب:**", bot_response)
+    st.write("**بوٹ کا جواب:**", bot_response)
+
+# "Try Yourself" section to modify AI model
+st.subheader("🤖 بوٹ کو اپنی مرضی کے مطابق بنائیں")
+new_question = st.text_input("نیا سوال شامل کریں:", "")
+new_answer = st.text_input("اس کا جواب شامل کریں:", "")
+
+if st.button("شامل کریں"):
+    if new_question and new_answer:
+        st.session_state.urdu_responses[new_question] = new_answer
+        st.success("کامیابی سے شامل ہو گیا!")
+    else:
+        st.warning("براہ کرم سوال اور جواب دونوں بھریں")
+
+# Display all Q&A
+st.subheader("📚 موجودہ سوالات اور جوابات")
+for q, a in st.session_state.urdu_responses.items():
+    st.write(f"**سوال:** {q}")
+    st.write(f"**جواب:** {a}")
+    st.write("---")