Create app.py

app.py

@@ -0,0 +1,144 @@
+import os
+import gdown
+import time
+import gradio as gr
+
+# Modern Imports
+from langchain_community.document_loaders import PyPDFLoader
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain_huggingface import HuggingFaceEmbeddings
+from langchain_community.vectorstores import FAISS
+from langchain_groq import ChatGroq
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.runnables import RunnablePassthrough
+from langchain_core.output_parsers import StrOutputParser
+
+# ==========================================
+# 1. SETUP & KEYS
+# ==========================================
+# On Hugging Face, you will set GROQ_API_KEY in the "Settings" tab under "Secrets"
+os.environ["GROQ_API_KEY"] = os.getenv('GROQ_API_KEY')
+
+# Assembly Language and Data Structures Knowledge Base
+links_to_process = [
+    "https://drive.google.com/file/d/1rb7AeJZrDNR-bq8Q9V4IvtzYZsDOvDH0/view?usp=sharing",
+    "https://drive.google.com/file/d/16PcJo_JaQHh1bx01lCAkc4QwQ6YnLb-K/view?usp=sharing"
+]
+
+output_dir = 'knowledge_base'
+if not os.path.exists(output_dir):
+    os.makedirs(output_dir)
+
+# ==========================================
+# 2. IMPROVED DOWNLOAD LOGIC
+# ==========================================
+def build_vector_db(links):
+    print(f"📥 Syncing Computer Science Knowledge Base...")
+    
+    for link in links:
+        try:
+            if "/folders/" in link:
+                gdown.download_folder(url=link, output=output_dir, quiet=True, use_cookies=False)
+            else:
+                gdown.download(url=link, output=output_dir + "/", quiet=True)
+            time.sleep(1) 
+        except Exception as e:
+            print(f"⚠️ Skip Link: {e}")
+
+    all_docs = []
+    for root, dirs, files in os.walk(output_dir):
+        for filename in files:
+            if filename.endswith(".pdf"):
+                file_path = os.path.join(root, filename)
+                try:
+                    loader = PyPDFLoader(file_path)
+                    all_docs.extend(loader.load())
+                except Exception:
+                    pass
+    
+    if not all_docs:
+        return None
+
+    text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=150)
+    chunks = text_splitter.split_documents(all_docs)
+    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+    return FAISS.from_documents(chunks, embeddings)
+
+# Initialize 
+vector_store = build_vector_db(links_to_process)
+if vector_store:
+    retriever = vector_store.as_retriever(search_kwargs={"k": 3})
+else:
+    retriever = None
+
+# ==========================================
+# 3. MODERN RAG CHAIN
+# ==========================================
+llm = ChatGroq(model="llama-3.3-70b-versatile", temperature=0)
+
+template = """You are a CS Professor's assistant. Answer the question about Assembly Language or Data Structures using the provided context. 
+If the answer is not in the context, say you don't know based on current documents.
+
+Context:
+{context}
+
+Question: {question}
+
+Expert Answer:"""
+
+prompt = ChatPromptTemplate.from_template(template)
+
+if retriever:
+    rag_chain = (
+        {"context": retriever, "question": RunnablePassthrough()}
+        | prompt
+        | llm
+        | StrOutputParser()
+    )
+else:
+    rag_chain = None
+
+# ==========================================
+# 4. FRONTEND
+# ==========================================
+custom_css = """
+#main-container { max-width: 900px; margin: auto; padding: 20px; }
+.header-text { text-align: center; color: #2563eb; margin-bottom: 2px; }
+.report-box { background-color: #f8fafc; border-radius: 12px; border: 1px solid #e2e8f0; padding: 20px; }
+"""
+
+def process_query(query):
+    if not rag_chain:
+        return "❌ Error: Knowledge Base not loaded. Check Google Drive links."
+    if not query.strip(): 
+        return "### ⚠️ Please enter a question regarding Assembly or Data Structures."
+    try:
+        return rag_chain.invoke(query)
+    except Exception as e:
+        return f"### ❌ Error\n{str(e)}"
+
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), css=custom_css) as demo:
+    with gr.Column(elem_id="main-container"):
+        gr.Markdown("# 🏛️ ByteWise: CS Fundamental Intelligence", elem_classes="header-text")
+        gr.Markdown("<p style='text-align: center;'>Specialized RAG for Assembly Language & Data Structures</p>")
+        gr.HTML("<hr>")
+        
+        user_input = gr.Textbox(
+            label="Inquiry", 
+            placeholder="e.g., Explain the stack operations in Assembly vs Linked List implementation...", 
+            lines=3
+        )
+        
+        with gr.Row():
+            submit_btn = gr.Button("CONSULT KNOWLEDGE BASE", variant="primary", scale=2)
+            clear_btn = gr.ClearButton([user_input], value="CLEAR", scale=1)
+
+        gr.Markdown("### 📋 Academic Intelligence Report")
+        with gr.Column(elem_classes="report-box"):
+            output_display = gr.Markdown(value="_Results will be rendered using academic context._")
+
+    submit_btn.click(fn=process_query, inputs=user_input, outputs=output_display)
+    user_input.submit(fn=process_query, inputs=user_input, outputs=output_display)
+
+if __name__ == "__main__":
+    demo.launch()