Update app.py

app.py

@@ -2,7 +2,7 @@ from flask import Flask, request, jsonify, render_template
 from flask_cors import CORS
 from dotenv import load_dotenv
 import os
-
+import re
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import Chroma
 from langchain_google_genai import ChatGoogleGenerativeAI
@@ -30,7 +30,7 @@ def get_qa_chain():
             google_api_key=GOOGLE_API_KEY,
             convert_system_message_to_human=True
         )
-
+        
         # Embeddings and vector store
         embedding_model = HuggingFaceEmbeddings(model_name="BAAI/bge-large-en-v1.5")
         vectordb = Chroma(
@@ -39,7 +39,7 @@ def get_qa_chain():
             collection_name="pdf_search_chroma"
         )
         retriever = vectordb.as_retriever(search_kwargs={"k": 6})
-
+        
         # Prompt
         prompt_template = PromptTemplate.from_template("""
         You are an intelligent assistant for students asking about their university.
@@ -52,7 +52,7 @@ def get_qa_chain():
         {question}
         Answer:
         """)
-
+        
         # Create chain
         qa_chain = RetrievalQA.from_chain_type(
             llm=llm,
@@ -62,6 +62,14 @@ def get_qa_chain():
         )
     return qa_chain
 
+def convert_links_to_html(text):
+    """
+    Convert links in the format [Text][URL] to HTML <a> tags.
+    Example: [Profile][https://example.com] becomes <a href="https://example.com">Profile</a>
+    """
+    pattern = r'\[(.*?)\]\[(.*?)\]'
+    return re.sub(pattern, r'<a href="\2">\1</a>', text)
+
 @app.route("/")
 def home():
     return render_template("index.html")
@@ -70,15 +78,18 @@ def home():
 def get_response():
     data = request.get_json()
     query = data.get("message", "")
+    
     if not query:
         return jsonify({"response": {"response": "No message received."}}), 400
-
+    
     chain = get_qa_chain()
     try:
         response = chain.run(query)
-        return jsonify({"response": {"response": response}})
+        # Convert links to HTML format
+        response_with_html_links = convert_links_to_html(response)
+        return jsonify({"response": {"response": response_with_html_links}})
     except Exception as e:
         return jsonify({"response": {"response": f"Error: {str(e)}"}}), 500
 
 if __name__ == "__main__":
-    app.run(debug=True)
+    app.run(debug=True)