Update app.py

app.py

@@ -75,28 +75,24 @@ def retrieve_and_generate_app(query, top_k=3):
     if not model or not index or not document_ids or not documents or not llm_pipeline:
         return "System not fully initialized. Please check logs for missing components."
 
-    # 1. Query Embedding
-    query_embedding = model.encode([query])
-
-    # 2. Retrieval using FAISS
-    D, I = index.search(query_embedding, top_k)
-
-    retrieved_docs = []
-    for i in I[0]:
-        sol_id = document_ids[i]
-        # Find the full content of the retrieved SOL
-        # --- CHANGE THIS LINE ---
-        # Original (incorrect): retrieved_content = next((doc["text"] for doc in documents if doc["id"] == sol_id), "Content not found.")
-        retrieved_content = next((doc["content"] for doc in documents if doc["id"] == sol_id), "Content not found.")
-        retrieved_docs.append({"id": sol_id, "content": retrieved_content})
-
-    # 3. Context Construction
-    # --- CHANGE THIS LINE ---
-    # Original (incorrect): context = "\n\n".join([f"SOL {doc['id']}: {doc['text']}" for doc in retrieved_docs])
-    context = "\n\n".join([f"SOL {doc['id']}: {doc['content']}" for doc in retrieved_docs])
-
-    # 4. LLM Generation
-    prompt = f"""
+    try:
+        # 1. Query Embedding
+        query_embedding = model.encode([query])
+
+        # 2. Retrieval using FAISS
+        D, I = index.search(query_embedding, top_k)
+
+        retrieved_docs = []
+        for i in I[0]:
+            sol_id = document_ids[i]
+            retrieved_content = next((doc["content"] for doc in documents if doc["id"] == sol_id), "Content not found.")
+            retrieved_docs.append({"id": sol_id, "content": retrieved_content})
+
+        # 3. Context Construction
+        context = "\n\n".join([f"SOL {doc['id']}: {doc['content']}" for doc in retrieved_docs])
+
+        # 4. LLM Generation
+        prompt = f"""
 Given the following information about Virginia Standards of Learning (SOLs):
 {context}
 Based on this information, answer the following question:
@@ -105,25 +101,28 @@ If the question is about a specific SOL number, provide a direct explanation for
 If asked for lesson plans, worksheets, or proofs, explain what the document generally entails and whether it provides such materials.
 Be concise and to the point.
 """
-    print(f"\n--- PROMPT SENT TO LLM ---\n{prompt}\n--------------------------\n")
+        print(f"\n--- PROMPT SENT TO LLM ---\n{prompt}\n--------------------------\n")
 
-    response = llm_pipeline(prompt, max_new_tokens=500, num_return_sequences=1, do_sample=True, temperature=0.7)
+        response = llm_pipeline(prompt, max_new_tokens=500, num_return_sequences=1, do_sample=True, temperature=0.7)
 
-    generated_text = response[0]['generated_text']
+        generated_text = response[0]['generated_text']
 
-    print(f"\n--- RAW GENERATED TEXT ---\n{generated_text}\n--------------------------\n")
+        print(f"\n--- RAW GENERATED TEXT ---\n{generated_text}\n--------------------------\n")
 
-    answer_start_marker = f"Based on this information, answer the following question:\n{query}"
-    if answer_start_marker in generated_text:
-        answer = generated_text.split(answer_start_marker, 1)[1].strip()
-        answer = re.sub(r'If the question is about a specific SOL number,.*?$', '', answer, flags=re.DOTALL).strip()
-    else:
-        answer = generated_text
+        answer_start_marker = f"Based on this information, answer the following question:\n{query}"
+        if answer_start_marker in generated_text:
+            answer = generated_text.split(answer_start_marker, 1)[1].strip()
+            answer = re.sub(r'If the question is about a specific SOL number,.*?$', '', answer, flags=re.DOTALL).strip()
+        else:
+            answer = generated_text
 
-    print(f"\n--- FINAL ANSWER ---\n{answer}\n--------------------\n")
-    return answer
+        print(f"\n--- FINAL ANSWER ---\n{answer}\n--------------------\n")
+        return answer if answer else "No valid response generated. Check logs for details."
+    except Exception as e:
+        print(f"\n--- ERROR ---\n{str(e)}\n------------\n")
+        return f"An error occurred: {str(e)}. Please check the logs for more details."
 
-    # Create Gradio interface
+# Create Gradio interface
 demo = gr.Interface(
     fn=retrieve_and_generate_app,
     inputs=gr.Textbox(lines=2, placeholder="Enter your geometry-related question here..."),
@@ -132,6 +131,5 @@ demo = gr.Interface(
     description="Ask questions about the Geometry SOL Instructional Guide"
 )
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch()