Update app.py

app.py

@@ -1,18 +1,15 @@
-# app.py
-
 import os
 from groq import Groq
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import FAISS
 import gradio as gr
 
-# Get GROQ_API_KEY from env
-groq_api_key = os.environ.get("GROQ_API_KEY")
-groq_client = Groq(api_key=groq_api_key)
+# Initialize Groq
+groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Load and embed documents
+# Load documents and create vector DB
 def load_documents_and_create_vectorstore():
     docs = []
     for file in ["documents/ASTM1.pdf", "documents/ASTM2.pdf"]:
@@ -27,22 +24,22 @@ def load_documents_and_create_vectorstore():
 
     return vectorstore
 
+# Load documents on startup
 vectorstore = load_documents_and_create_vectorstore()
 
-# RAG: Ask question using context + Groq
+# RAG question answering function
 def ask_question(question):
     retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
     docs = retriever.get_relevant_documents(question)
 
     context = "\n".join([doc.page_content for doc in docs])
+    prompt = f"""You are a helpful Civil Engineering assistant. Use the following ASTM standard context to answer:
 
-    prompt = f"""You are a helpful Civil Engineering assistant. Use the ASTM standard context below to answer:
-
-    Context:
-    {context}
+Context:
+{context}
 
-    Question: {question}
-    Answer:"""
+Question: {question}
+Answer:"""
 
     completion = groq_client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
@@ -57,5 +54,5 @@ gr.Interface(
     inputs=gr.Textbox(label="Ask a Civil Engineering Question (based on ASTM)"),
     outputs=gr.Textbox(label="Answer"),
     title="Civil Engineering RAG Assistant",
-    description="Ask any question about ASTM Civil Engineering Standards"
+    description="Ask questions from uploaded ASTM PDFs"
 ).launch()