Update app.py

app.py

@@ -24,14 +24,14 @@ def search_and_answer(query):
         return "No relevant documents found."
     count = 0
     for doc in retrieved_docs:
-      count += 1
-      
-    print(f"Related Courses:{count}")
-    course_info = [(doc.metadata.get('title', 'No Title'), doc.metadata.get('lessons', 'courses')) for doc in retrieved_docs]    
+        count += 1
+    print(f"Related Courses: {count}")
+    course_info = [(doc.metadata.get('title', 'No Title'), doc.metadata.get('lessons', 'No lessons')) for doc in retrieved_docs]    
     if course_info:
-        return '\n'.join([f"Course ID: {course_id}\nCourse Title: {course_title}\n" for course_title, course_id in course_info])
+        return '\n'.join([f"Course Title: {course_title}\nCourse Lessons: {course_lessons}\n" for course_title, course_lessons in course_info])
     else:
         return "No relevant courses found."
+
 st.title("Course Search WebApp")
 st.write("Enter a query to search for related courses:")
 
@@ -40,11 +40,8 @@ query = st.text_input("Query")
 
 if query:
     # Get results
-    context, answer = search_and_answer(query)
+    result = search_and_answer(query)
 
     # Display the results
     st.subheader("Answer:")
-    st.write(answer)
-
-    st.subheader("Relevant Course Titles:")
-    st.write(context)
+    st.write(result)