Update app.py

app.py

@@ -1,28 +1,38 @@
 import os
 import gradio as gr
 from groq import Groq
-from notion_client import Client
+import requests
 
 groq_client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
 )
 
-notion_client = Client(auth=os.environ.get("NOTION_TOKEN"))
-notion_database_id = "4fc0a081f0a84257879d6f7638e368b9"
+NOTION_TOKEN = os.environ.get("NOTION_TOKEN")
+NOTION_DATABASE_ID = "4fc0a081f0a84257879d6f7638e368b9"
 
 def store_conversation(user_input, bot_response):
     # Truncate bot_response if it exceeds 2000 characters
     truncated_response = bot_response[:1997] + "..." if len(bot_response) > 2000 else bot_response
+    
+    url = f"https://api.notion.com/v1/pages"
+    headers = {
+        "Authorization": f"Bearer {NOTION_TOKEN}",
+        "Content-Type": "application/json",
+        "Notion-Version": "2022-06-28"
+    }
+    data = {
+        "parent": {"database_id": NOTION_DATABASE_ID},
+        "properties": {
+            "User Input": {"title": [{"text": {"content": user_input}}]},
+            "Bot Response": {"rich_text": [{"text": {"content": truncated_response}}]}
+        }
+    }
+    
     try:
-        notion_client.pages.create(
-            parent={"database_id": notion_database_id},
-            properties={
-                "User Input": {"title": [{"text": {"content": user_input}}]},
-                "Bot Response": {"rich_text": [{"text": {"content": truncated_response}}]}
-            }
-        )
+        response = requests.post(url, headers=headers, json=data)
+        response.raise_for_status()
         print("Conversation stored successfully")
-    except Exception as e:
+    except requests.exceptions.RequestException as e:
         print(f"Error storing conversation: {str(e)}")
         print("Make sure the Notion database is shared with your integration and the database ID is correct.")
 
@@ -50,6 +60,6 @@ demo = gr.ChatInterface(fn=chat_with_groq,
                         clear_btn="Clear",)
 
 if __name__ == "__main__":
-    print(f"Using Notion database ID: {notion_database_id}")
+    print(f"Using Notion database ID: {NOTION_DATABASE_ID}")
     print("Make sure the Notion database is shared with your integration.")
     demo.launch()