Update app.py

app.py

@@ -9,16 +9,19 @@ database_str=''
 with open('data_base.txt', 'r',encoding='utf-8') as f:
     database_str = f.read()
 
-
 def generate_response(query):
     prompt = f'''
-    You are a Course suggestor based on the user requirement and the from the given database which consist of 
-    the course name and description of the course.
+    You are a Course suggestor based on the user requirement and the given database, which consists of 
+    the course name, description, and link to each course.
 
-    You're tasked to use the description of each course and compare it with the user input and output which course's
-    description matches the user requirement.
-    Output the course name & Course Link alone which matches the user requirement.
-    you may output a max of 3 courses if you find that are good matches. name of the course should be exactly same as the database provided to you along with its link provided
+    You're tasked to use the description of each course and compare it with the user input, then output the courses
+    whose description matches the user requirement.
+    
+    Strictly output the course name and its corresponding link, following the exact format below:
+    - Output a maximum of 3 courses if they match well.
+    - The format should be exactly: Course Name || Course Link
+    - Each course should be on a new line.
+    - No extra text or commentary, only the exact output format specified.
 
     # Database
     {database_str}
@@ -26,11 +29,12 @@ def generate_response(query):
     # User Input
     {query}
 
-    # Output : Course Name||Coure LINK \ Course Name||Course LINK \....
+    # Output (maximum of 3 courses):
     '''
     model = genai.GenerativeModel("gemini-1.5-flash")
     response = model.generate_content(prompt)
-    return response.text.split("\\")
+    return response.text.strip().split("\n")
+
 
 
 # Define session state variables
@@ -63,9 +67,21 @@ for message in st.session_state.messages:
             st.write(message['content'])
     else:
         with st.chat_message("ai"):
-            for i in message['content']:
-                name = i.split('||')[0]
-                link = i.split("||")[1]
-                st.markdown(f"[{name}]({link})", unsafe_allow_html=True)
+            for item in message['content']:
+                # Skip if the item is empty
+                if not item.strip():
+                    continue
+
+                # Attempt to split the item by '||'
+                try:
+                    name, link = item.split('||')
+                    # Strip any extra whitespace
+                    name = name.strip()
+                    link = link.strip()
 
-                
+                    # Display as markdown with link
+                    st.markdown(f"[{name}]({link})", unsafe_allow_html=True)
+                except ValueError:
+                    # If the item does not match the expected format, provide a warning message
+                    # st.write("Warning: Unable to parse the course recommendation properly.")
+                    print("Warning")