Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 from groq import Groq
 import requests
+import re
 
 groq_client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
@@ -10,6 +11,52 @@ groq_client = Groq(
 NOTION_TOKEN = os.environ.get("NOTION_TOKEN")
 NOTION_PAGE_ID = "4fc0a081f0a84257879d6f7638e368b9"  # Replace with your actual page ID
 
+def format_text_for_notion(text):
+    # Split the text into lines
+    lines = text.split('\n')
+    formatted_text = []
+
+    for line in lines:
+        # Check if the line is a bullet point
+        if line.strip().startswith('•') or line.strip().startswith('-'):
+            formatted_text.append({
+                "type": "bulleted_list_item",
+                "bulleted_list_item": {
+                    "rich_text": [{"type": "text", "text": {"content": line.strip()[2:].strip()}}]
+                }
+            })
+        else:
+            # Check for bold text (assuming it's wrapped in ** or __)
+            bold_parts = re.split(r'(\*\*.*?\*\*|__.*?__)', line)
+            rich_text = []
+            for part in bold_parts:
+                if part.startswith('**') and part.endswith('**'):
+                    rich_text.append({
+                        "type": "text",
+                        "text": {"content": part[2:-2]},
+                        "annotations": {"bold": True}
+                    })
+                elif part.startswith('__') and part.endswith('__'):
+                    rich_text.append({
+                        "type": "text",
+                        "text": {"content": part[2:-2]},
+                        "annotations": {"bold": True}
+                    })
+                elif part:
+                    rich_text.append({
+                        "type": "text",
+                        "text": {"content": part}
+                    })
+            
+            formatted_text.append({
+                "type": "paragraph",
+                "paragraph": {
+                    "rich_text": rich_text
+                }
+            })
+
+    return formatted_text
+
 def store_conversation(user_input, bot_response):
     url = f"https://api.notion.com/v1/blocks/{NOTION_PAGE_ID}/children"
     headers = {
@@ -17,6 +64,9 @@ def store_conversation(user_input, bot_response):
         "Content-Type": "application/json",
         "Notion-Version": "2022-06-28"
     }
+    
+    formatted_bot_response = format_text_for_notion(bot_response)
+    
     data = {
         "children": [
             {
@@ -41,9 +91,16 @@ def store_conversation(user_input, bot_response):
                 "object": "block",
                 "type": "paragraph",
                 "paragraph": {
-                    "rich_text": [{"type": "text", "text": {"content": f"Bot: {bot_response[:1997]}..."}}]
+                    "rich_text": [
+                        {
+                            "type": "text",
+                            "text": {"content": "Bot: "},
+                            "annotations": {"bold": True}
+                        }
+                    ]
                 }
-            }
+            },
+            *formatted_bot_response
         ]
     }
     
@@ -74,7 +131,8 @@ def chat_with_groq(user_input, additional_context=None):
 demo = gr.ChatInterface(fn=chat_with_groq, 
                         textbox=gr.Textbox(placeholder="Ask me any question"), 
                         title="Hey NOPE", theme="Monochrome", 
-                        description="Welcome to the world of NOPE",
+                        description="Welcome to the world of NOPE", 
+                        examples=["Need some content Idea", "Generate some Thumbnail Text"], 
                         retry_btn=None, 
                         undo_btn="Delete Previous", 
                         clear_btn="Clear",)