Spaces:

Redfire-1234
/

PCB_Tutor

Sleeping

App Files Files Community

Redfire-1234 commited on Dec 7, 2025

Commit

3391f7c

verified ·

1 Parent(s): aa8087f

Create app.py

Browse files

Files changed (1) hide show

app.py +194 -0

app.py ADDED Viewed

	@@ -0,0 +1,194 @@

+from flask import Flask, request, jsonify, render_template_string
+from sentence_transformers import SentenceTransformer
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import hf_hub_download
+import faiss
+import pickle
+import torch
+app = Flask(__name__)
+print("Loading models and data...")
+# Download FAISS index and chunks
+faiss_path = hf_hub_download(
+    repo_id="Redfire-1234/PCB",
+    filename="faiss_index.bin",
+    repo_type="dataset"
+)
+chunks_path = hf_hub_download(
+    repo_id="Redfire-1234/PCB",
+    filename="all_chunks.pkl",
+    repo_type="dataset"
+)
+# Load FAISS and chunks
+index = faiss.read_index(faiss_path)
+with open(chunks_path, "rb") as f:
+    all_chunks = pickle.load(f)
+# Load models
+embed_model = SentenceTransformer("all-MiniLM-L6-v2")
+model_name = "Qwen/Qwen2.5-1.5B-Instruct"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
+print(f"All models loaded! Device: {device}")
+def search(query, k=3):
+    q_emb = embed_model.encode([query]).astype("float32")
+    D, I = index.search(q_emb, k)
+    return "\n".join([all_chunks[i] for i in I[0]])
+def generate_answer(query, context, mode="answer"):
+    if mode == "mcq":
+        prompt = f"""Generate 1 MCQ from this text:
+- 4 options (A, B, C, D)
+- Mark correct answer
+Text: {context}
+Question:"""
+    else:
+        prompt = f"""Answer based on this text:
+Text: {context}
+Question: {query}
+Answer:"""
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=1024).to(device)
+    outputs = model.generate(**inputs, max_new_tokens=200)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+HTML_TEMPLATE = """
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Class 12 PCB AI Tutor</title>
+    <style>
+        body { font-family: Arial; max-width: 800px; margin: 50px auto; padding: 20px; background: #f5f5f5; }
+        .container { background: white; padding: 40px; border-radius: 15px; box-shadow: 0 5px 20px rgba(0,0,0,0.1); }
+        h1 { color: #6366f1; text-align: center; margin-bottom: 10px; }
+        p { text-align: center; color: #666; margin-bottom: 30px; }
+        textarea { width: 100%; padding: 15px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; margin-bottom: 15px; min-height: 100px; font-family: inherit; }
+        button { width: 100%; padding: 15px; background: #6366f1; color: white; border: none; border-radius: 8px; font-size: 16px; cursor: pointer; font-weight: 600; }
+        button:hover { background: #4f46e5; }
+        button:disabled { background: #ccc; cursor: not-allowed; }
+        .tabs { display: flex; gap: 10px; margin-bottom: 20px; }
+        .tab { flex: 1; padding: 12px; background: #f0f0f0; border: none; cursor: pointer; border-radius: 8px; font-weight: 600; }
+        .tab.active { background: #6366f1; color: white; }
+        .tab-content { display: none; }
+        .tab-content.active { display: block; }
+        .result { margin-top: 25px; padding: 20px; background: #f9fafb; border-radius: 8px; border-left: 4px solid #6366f1; display: none; }
+        .result.show { display: block; }
+        .result h3 { color: #6366f1; margin-bottom: 15px; }
+        .result .answer { background: white; padding: 15px; border-radius: 6px; line-height: 1.6; white-space: pre-wrap; }
+        .loading { text-align: center; margin-top: 15px; color: #666; display: none; }
+        .loading.show { display: block; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>🎓 Class 12 PCB AI Tutor</h1>
+        <p>Ask questions or generate MCQs from Physics, Chemistry & Biology</p>
+        <div class="tabs">
+            <button class="tab active" onclick="switchTab('answer')">Ask Question</button>
+            <button class="tab" onclick="switchTab('mcq')">Generate MCQ</button>
+        </div>
+        <div id="answer-tab" class="tab-content active">
+            <textarea id="queryInput" placeholder="Ask any question from Class 12 PCB..."></textarea>
+            <button onclick="getAnswer()">Get Answer</button>
+        </div>
+        <div id="mcq-tab" class="tab-content">
+            <textarea id="mcqTopic" placeholder="Enter topic for MCQ generation..."></textarea>
+            <button onclick="generateMCQ()">Generate MCQ</button>
+        </div>
+        <div class="loading" id="loading">Processing...</div>
+        <div class="result" id="result">
+            <h3 id="resultTitle">Answer:</h3>
+            <div class="answer" id="answer"></div>
+        </div>
+    </div>
+    <script>
+        function switchTab(tab) {
+            document.querySelectorAll('.tab').forEach(t => t.classList.remove('active'));
+            document.querySelectorAll('.tab-content').forEach(c => c.classList.remove('active'));
+            event.target.classList.add('active');
+            document.getElementById(tab + '-tab').classList.add('active');
+            document.getElementById('result').classList.remove('show');
+        }
+        async function getAnswer() {
+            const query = document.getElementById('queryInput').value.trim();
+            if (!query) { alert('Please enter a question!'); return; }
+            await makeRequest(query, 'answer');
+        }
+        async function generateMCQ() {
+            const topic = document.getElementById('mcqTopic').value.trim();
+            if (!topic) { alert('Please enter a topic!'); return; }
+            await makeRequest(topic, 'mcq');
+        }
+        async function makeRequest(query, mode) {
+            const loading = document.getElementById('loading');
+            const result = document.getElementById('result');
+            loading.classList.add('show');
+            result.classList.remove('show');
+            try {
+                const response = await fetch('/query', {
+                    method: 'POST',
+                    headers: {'Content-Type': 'application/json'},
+                    body: JSON.stringify({query, mode})
+                });
+                const data = await response.json();
+                if (data.error) { alert('Error: ' + data.error); return; }
+                document.getElementById('resultTitle').textContent = mode === 'mcq' ? 'Generated MCQ:' : 'Answer:';
+                document.getElementById('answer').textContent = data.answer;
+                result.classList.add('show');
+            } catch (error) {
+                alert('Error: ' + error.message);
+            } finally {
+                loading.classList.remove('show');
+            }
+        }
+    </script>
+</body>
+</html>
+"""
+@app.route('/')
+def home():
+    return render_template_string(HTML_TEMPLATE)
+@app.route('/query', methods=['POST'])
+def query():
+    try:
+        data = request.get_json()
+        if not data or 'query' not in data:
+            return jsonify({'error': 'No query provided'}), 400
+        query_text = data['query']
+        mode = data.get('mode', 'answer')
+        context = search(query_text, k=3)
+        answer = generate_answer(query_text, context, mode)
+        return jsonify({'query': query_text, 'answer': answer, 'mode': mode})
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+@app.route('/health')
+def health():
+    return jsonify({'status': 'healthy'})
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)