Update app.py

app.py

@@ -1,23 +1,25 @@
 import streamlit as st
 from transformers import pipeline
 
-# Title
-st.title("🤖 AI Chatbot - Hugging Face")
+st.title("❓ AI Question Answering Bot")
 
-# Initialize chatbot model
 @st.cache_resource
-def load_chatbot():
-    return pipeline("text-generation", model="microsoft/DialoGPT-medium")
+def load_qa():
+    return pipeline("question-answering", model="deepset/roberta-base-squad2")
 
-chatbot = load_chatbot()
+qa = load_qa()
 
-# User input
-user_input = st.text_input("You:", "")
+# Set a static context or let user provide it
+context = st.text_area("📄 Paste your reference text (context):", 
+                       "The QA QC Electrical Inspector ensures the quality of installations, performs inspection and testing, and checks documentation in power and oil & gas projects.")
 
-# If user types something
-if user_input:
-    # Generate response
-    with st.spinner("Chatbot is thinking..."):
-        response = chatbot(user_input, max_length=100, do_sample=True, pad_token_id=50256)
-        bot_reply = response[0]['generated_text'].split(user_input)[-1].strip()
-        st.markdown(f"**Bot:** {bot_reply}")
+question = st.text_input("❓ Your question:")
+
+if question:
+    try:
+        with st.spinner("Thinking..."):
+            answer = qa(question=question, context=context)
+            st.markdown(f"**Answer:** {answer['answer']}")
+    except Exception as e:
+        st.error("Something went wrong.")
+        st.exception(e)