Update app.py

app.py

@@ -1,7 +1,29 @@
 import gradio as gr
 import pandas as pd
 import random
+import os
+from datetime import datetime
+import requests
 
+# Initialize Groq Chatbot
+try:
+    from groq import Groq
+except ImportError:
+    os.system('pip install groq')
+    from groq import Groq
+
+groq_api_key = os.getenv("groq_key")
+notion_api_key = os.getenv("Notion_API_Key")
+notion_db_id = os.getenv("Notion_DB_ID")
+
+if not groq_api_key:
+    raise ValueError("'groq_key' is missing. Please set the API key.")
+if not notion_api_key or not notion_db_id:
+    raise ValueError("Notion API Key or Database ID is missing.")
+
+client = Groq(api_key=groq_api_key)
+
+# Vocabulary App Functions
 vocabulary = []
 matches = {}
 score = 0
@@ -47,16 +69,60 @@ def reset_quiz():
     score = 0
     return "Quiz reset! Ready for a new start."
 
+# Groq Chatbot Integration
+def chat_with_groq(input_text, chat_history):
+    try:
+        messages = [
+            {"role": "system", "content": (
+                "You are an English dictionary and quiz designer tutor. When given a vocabulary word, 
+                you should provide the following: \n(1) A concise definition (max 50 words)\n(2) A sample sentence (max 20 words)."
+            )}
+        ]
+        messages.extend(
+            {"role": "user", "content": user_msg} if idx % 2 == 0 else {"role": "assistant", "content": assistant_msg}
+            for idx, (user_msg, assistant_msg) in enumerate(chat_history)
+        )
+        messages.append({"role": "user", "content": input_text})
+
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=messages,
+            temperature=1,
+            max_tokens=1024,
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Error: {e}"
+
+def log_to_notion(name, user_input, bot_response):
+    url = "https://api.notion.com/v1/pages"
+    headers = {
+        "Authorization": f"Bearer {notion_api_key}",
+        "Content-Type": "application/json",
+        "Notion-Version": "2022-06-28"
+    }
+    data = {
+        "parent": {"database_id": notion_db_id},
+        "properties": {
+            "Name": {"title": [{"text": {"content": name}}]},
+            "Timestamp": {"date": {"start": datetime.now().isoformat()}},
+            "User Input": {"rich_text": [{"text": {"content": user_input}}]},
+            "Bot Response": {"rich_text": [{"text": {"content": bot_response}}]}
+        }
+    }
+    requests.post(url, headers=headers, json=data)
+
+# Gradio UI
 with gr.Blocks() as app:
     with gr.Row():
-        gr.Markdown("# 📚 Personal English Vocabulary App")
+        gr.Markdown("# 📚 Personal English Vocabulary App & Chatbot")
 
+    # Tabs
     with gr.Tab("Upload Vocabulary"):
         with gr.Row():
             file_input = gr.File(label="Upload Vocabulary CSV")
             file_output = gr.Textbox(label="Upload Status")
             upload_button = gr.Button("Upload")
-
         upload_button.click(parse_csv, inputs=file_input, outputs=file_output)
 
     with gr.Tab("Add Vocabulary"):
@@ -64,13 +130,12 @@ with gr.Blocks() as app:
         definition_input = gr.Textbox(label="Definition")
         add_button = gr.Button("Add Word")
         add_status = gr.Textbox(label="Status")
-
         add_button.click(add_word, inputs=[word_input, definition_input], outputs=add_status)
 
     with gr.Tab("Quiz"):
         quiz_status = gr.Textbox(label="Status")
         start_button = gr.Button("Start Quiz")
-        words_list = gr.Dropdown(label="Select Word", choices=[])  # Initialize dropdowns as empty
+        words_list = gr.Dropdown(label="Select Word", choices=[])
         definitions_list = gr.Dropdown(label="Select Definition", choices=[])
         check_button = gr.Button("Check Match")
         match_status = gr.Textbox(label="Result")
@@ -82,10 +147,25 @@ with gr.Blocks() as app:
         start_button.click(refresh_quiz, inputs=None, outputs=[quiz_status, words_list, definitions_list])
         check_button.click(check_match, inputs=[words_list, definitions_list], outputs=match_status)
 
+    with gr.Tab("Vocabulary Chatbot"):
+        gr.Markdown("## AI-Powered Vocabulary Chatbot")
+        chatbot = gr.Chatbot()
+        user_input = gr.Textbox(label="Ask about a word or usage:")
+        name_input = gr.Textbox(label="Name", placeholder="Enter your name")
+        send_button = gr.Button("Submit")
+        chat_history = gr.State([])
+
+        def respond(name, message, chat_history):
+            bot_response = chat_with_groq(message, chat_history)
+            chat_history.append((message, bot_response))
+            log_to_notion(name, message, bot_response)
+            return chat_history, ""
+
+        send_button.click(respond, [name_input, user_input, chat_history], [chatbot, user_input])
+
     with gr.Row():
         reset_button = gr.Button("Reset Quiz")
         reset_status = gr.Textbox(label="Reset Status")
-
         reset_button.click(reset_quiz, inputs=None, outputs=reset_status)
 
 app.launch()