Update app.py

app.py

@@ -7,7 +7,11 @@ import faiss
 import nltk
 from groq import Groq
 
-nltk.download('punkt')
+# Ensure the punkt tokenizer is downloaded
+try:
+    nltk.data.find('tokenizers/punkt')
+except LookupError:
+    nltk.download('punkt')
 
 # Initialize Groq client
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
@@ -96,18 +100,4 @@ def main():
         st.header("Ask Questions")
         query = st.text_input("Enter your question here")
         if st.button("Query Document"):
-            results = query_faiss(st.session_state['faiss_index'], query, st.session_state['chunks'], st.session_state['model'])
-            st.write("### Results from Document:")
-            for i, result in enumerate(results):
-                st.write(f"**Result {i+1}:** {result}")
-
-            # Use Groq API for additional insights
-            chat_completion = client.chat.completions.create(
-                messages=[{"role": "user", "content": query}],
-                model="llama-3.3-70b-versatile",
-            )
-            st.write("### Insights from Groq-powered Model:")
-            st.write(chat_completion.choices[0].message.content)
-
-if __name__ == "__main__":
-    main()
+            results = query_faiss(st.session_state['faiss_index'],