Update app.py

app.py

@@ -19,7 +19,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)
+    answer = t5_pipeline(input_text, max_length = 300,truncation = True)
     return answer[0]['generated_text']
 
     
@@ -65,6 +65,7 @@ def main():
     query = st.text_input("🔍 Enter your query:")
 
     if st.button("Search"):
+        with st.spinner("Fetching results..."):
         start_time = time.time()
         response, course_info, summary = search_and_answer(query, index_file="faiss_course_index.pkl")
         
@@ -76,12 +77,12 @@ def main():
             
             # Display related courses as cards
             st.subheader("🎓 Related Courses:")
-            st.write("<div style='display: flex; flex-wrap: wrap; justify-content: center;'>", unsafe_allow_html=True)
+            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: 200px; padding: 10px; margin: 10px; border: 1px solid #ddd; border-radius: 8px; box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1);'>
-                    <h3 style='font-size: 16px; color: #333;'>{course_title}</h3>
-                    <p style='font-size: 14px; color: #666;'>Lessons: {course_lessons}</p>
+                <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>
                 """
                 st.write(card_html, unsafe_allow_html=True)
@@ -89,10 +90,9 @@ def main():
 
             # Display summary with better layout
             st.subheader("📝 Summary:")
-            st.info(summary)
+            st.info(summary[:300])
 
     st.markdown("<p style='text-align: center;'>Developed with ❤️ by Prachi</p>", unsafe_allow_html=True)
 
 if __name__ == "__main__":
-    main()
-
+    main()