import streamlit as st
from transformers import AutoTokenizer, AutoModelForCausalLM

st.title("🌙 Noor-e-Hidayat – Islamic AI Chatbot")

# Load model
model_id = "Ellbendls/Qwen3-4b-Quran-LoRA-Fine-Tuned"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

# Input
user_input = st.text_input("🕊️ Ask a question based on Qur’an or Hadith:")

if user_input:
    st.write("🤖 Generating answer...")
    
    prompt = f"Answer the following with Qur’an-based reasoning:\nQuestion: {user_input}\nAnswer:"
    inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
    outputs = model.generate(**inputs, max_new_tokens=300)
    
    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
    
    st.markdown("### 📜 Answer:")
    st.write(response)