Update pages/🤖 Medical Question Answering.py

pages/🤖 Medical Question Answering.py

@@ -67,19 +67,10 @@ def get_conversational_chain():
 # Function to handle user input
 # Function to handle user input
 def user_input(user_question):
-    # embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
-    # new_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
-    # docs = new_db.similarity_search(user_question)
-    # chain = get_conversational_chain()
-    # response = chain({"input_documents": docs, "question": user_question}, return_only_outputs=True)
-
-    # return response["output_text"]  # Return the answer as a string
-    chain = get_conversational_chain()
-    response = chain({"input_documents": [], "question": user_question}, return_only_outputs=True)
-    
-    return response["output_text"]
-
-
+    model = genai.GenerativeModel('gemini-1.5-pro')
+    chat = model.start_chat(history=[])
+    response = chat.send_message(user_question)
+    return response.text
 
 # Streamlit UI setup
 st.markdown("<h1 style='text-align: center;'>Chào mừng tới Medical Question Answering 🎈</h1>", unsafe_allow_html=True)