Update app.py

app.py

@@ -1,17 +1,25 @@
 import streamlit as st
-import json
+from pymongo import MongoClient
 from gradio_client import Client
 import time
+import os
 
-# Load data.json
-with open('data.json') as f:
-    data = json.load(f)
+# Connect to MongoDB
+password = os.getenv("password")
+uri = f"mongodb+srv://E:{password}@cluster0.rvt8psd.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"
+client = MongoClient(uri)
+db = client["instantqa"]
+collection = db["instantqa"]
 
 client_endpoint = "https://olivier-truong-mistral-super-fast.hf.space/"
 
 def ask_question(question):
-    if question in data:
-        return data[question]
+    # Check if the question exists in the database
+    query = {"question": question}
+    result = collection.find_one(query)
+
+    if result:
+        return result["answer"]
     else:
         client = Client(client_endpoint)
         result = client.predict(
@@ -24,9 +32,11 @@ def ask_question(question):
         )
         answer = result
         answer = answer[:-4]
-        data[question] = answer
-        with open('data.json', 'w') as f:
-            json.dump(data, f)
+        
+        # Store the question-answer pair in MongoDB
+        data = {"question": question, "answer": answer}
+        collection.insert_one(data)
+
         return answer
 
 def typewriter(text: str, speed: int):
@@ -39,7 +49,7 @@ def typewriter(text: str, speed: int):
 
 def main():
     st.title("Near Instant Question Answering")
-    st.write("Limitations: May generate unhelpfull or outdated content, no chat history.")
+    st.write("Limitations: May generate unhelpful or outdated content, no chat history.")
 
     question = st.text_input("Ask your question:")
 
@@ -55,3 +65,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+