Update app.py

app.py

@@ -7,19 +7,23 @@ groq_client = Groq(
     api_key=os.environ.get("GROQ_API_KEY"),
 )
 
-notion_client = Client(auth=os.environ.get("NOTION_API_KEY"))
+notion_client = Client(auth=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
-    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}}]}
-        }
-    )
+    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}}]}
+            }
+        )
+        print("Conversation stored successfully")
+    except Exception as e:
+        print(f"Error storing conversation: {str(e)}")
 
 def chat_with_groq(user_input, additional_context=None):
     chat_completion = groq_client.chat.completions.create(
@@ -32,10 +36,7 @@ def chat_with_groq(user_input, additional_context=None):
         model="llama-3.1-8b-instant",
     )
     bot_response = chat_completion.choices[0].message.content
-    try:
-        store_conversation(user_input, bot_response)
-    except Exception as e:
-        print(f"Error storing conversation: {str(e)}")
+    store_conversation(user_input, bot_response)
     return bot_response
 
 demo = gr.ChatInterface(fn=chat_with_groq,