Update app.py

app.py

@@ -5,6 +5,8 @@ from langchain.schema import Document
 from transformers import pipeline
 import pickle
 import time
+from string import Template
+
 
 def load_faiss_index(index_file="faiss_course_index.pkl"):
     with open(index_file, "rb") as f:
@@ -19,7 +21,7 @@ def get_faiss_retriever(index_file="faiss_course_index.pkl"):
 def get_answer_from_t5(query, context):
     t5_pipeline = pipeline("text2text-generation", model="t5-small", device=0)
     input_text = f"Question: {query} Context: {context}"
-    answer = t5_pipeline(input_text, max_length = 300,truncation = True)
+    answer = t5_pipeline(input_text, max_length = 500,truncation = True)
     return answer[0]['generated_text']
 
     
@@ -46,19 +48,9 @@ def search_and_answer(query, index_file="faiss_course_index.pkl"):
 
     return response, course_info, summary
 
-import streamlit as st
-import time
-
-def create_card(title, lessons, image_url="https://via.placeholder.com/300x150", rating="4.0"):
-    card_template = f"""
-    <div style="width: 300px; margin: 10px; padding: 15px; background-color: #f9f9f9; 
-                border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.1); text-align: center;">
-        <img src="{image_url}" alt="Course Image" style="width: 100%; border-radius: 8px; margin-bottom: 10px;">
-        <h4 style="margin: 10px 0;">{title}</h4>
-        <p style="margin: 5px 0; font-size: 14px;">⭐ {rating} | {lessons} Lessons</p>
-    </div>
-    """
-    return card_template
+def load_card_template():
+    with open("card_template.html",'r') as file:
+        return file.read()
 
 def main():
     st.title("📚 Course Search WebApp")
@@ -75,22 +67,17 @@ def main():
             st.success("✅ Results fetched successfully!")
             st.write(f"⏱ **Time Taken:** {time.time() - start_time:.2f} seconds")
             
-            # Display related courses as cards
+            card_template = load_card_template()
             st.subheader("🎓 Related Courses:")
             st.write("<div style='display: flex; flex-wrap: wrap; justify-content: space-between;'>", unsafe_allow_html=True)
             for course_title, course_lessons in course_info:
-                card_html = f"""
-                <div style='width: 48%; padding: 10px; margin: 10px; border: 1px solid #f7f7f7; border-radius: 8px; box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1);'>
-                    <h3 style='font-size: 16px; color: #edebeb;'>{course_title}</h3>
-                    <p style='font-size: 14px; color: #c9c9c9;'>Lessons: {course_lessons}</p>
-                </div>
-                """
+               card_html = Template(card_template).substitute(course_title = course_title, course_lessons = course_lessons)
                 st.write(card_html, unsafe_allow_html=True)
             st.write("</div>", unsafe_allow_html=True)
 
             # Display summary with better layout
             st.subheader("📝 Summary:")
-            st.info(summary[:300])
+            st.info(summary[:500])
 
     st.markdown("<p style='text-align: center;'>Developed with ❤️ by Prachi</p>", unsafe_allow_html=True)