import streamlit as st

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 ہے۔",
        "پاکستان کا قائد اعظم کون ہے": "پاکستان کے قائد اعظم محمد علی جناح ہیں۔",
        "سر مکدم اچھے ہیں": "ہاں، وہ دنیا کے بہترین سر ہیں۔",  # New question and response
    }

# Streamlit app title
st.title("🕌 اردو زبان میں سادہ چیٹ بوٹ")

# Main chat interface
user_input = st.text_input("اپنا سوال یہاں درج کریں:", "")

def get_response(user_input):
    # Process input
    processed_input = user_input.strip().lower().replace('؟', '').replace('?', '')
    
    # Check for matching questions
    for question, response in st.session_state.urdu_responses.items():
        if processed_input == question.lower().replace('؟', ''):
            return response
    
    return "معاف کیجئے، میں آپ کے سوال کا جواب نہیں دے سکتا۔"

# Display conversation
if user_input:
    st.write("**آپ کا سوال:**", user_input)
    bot_response = get_response(user_input)
    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("---")