Update app.py

app.py

@@ -4,10 +4,23 @@ from langchain_google_genai import GoogleGenerativeAIEmbeddings
 import google.generativeai as genai
 from langchain_community.vectorstores import FAISS
 from langchain_google_genai import ChatGoogleGenerativeAI
+import re
+
 
 genai.configure(api_key="AIzaSyD2o8vjePJb6z8vT_PVe82lVWMD3_cBL0g")
 
 
+def format_gemini_response(text):
+
+  bold_pattern = r"\*\*(.*?)\*\*"
+  italic_pattern = r"\*(.*?)\*"
+  code_pattern = r"```(.*?)```"
+  text = text.replace('\n', '<br>')
+  formatted_text = re.sub(code_pattern,"<pre><code>\\1</code></pre>",text)
+  formatted_text = re.sub(bold_pattern, "<b>\\1</b>", formatted_text)
+  formatted_text = re.sub(italic_pattern, "<i>\\1</i>", formatted_text)
+
+  return formatted_text
 def predict(message :str ) -> str:
     model = genai.GenerativeModel("gemini-pro")
     his = []
@@ -20,7 +33,7 @@ def predict(message :str ) -> str:
         history=his
     )
     response = chat.send_message(message)
-    return response.text
+    return format_gemini_response(response.text)
 iface = gr.Interface(fn = predict,inputs = ["text"],outputs = "text")
 iface.launch()