Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import random
 import pandas as pd
 import ssl
+import os
+from datetime import datetime
+import requests
 
 # Workaround for SSL issues in some environments
 ssl._create_default_https_context = ssl._create_unverified_context
@@ -12,7 +15,24 @@ matches = {}
 score = 0
 remaining_pairs = 0
 
-# Helper Functions
+# Initialize Groq and Notion APIs
+try:
+    from groq import Groq
+except ImportError:
+    os.system('pip install groq')
+    from groq import 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 api_key:
+    raise ValueError("Environment variable 'groq_key' not set.")
+if not notion_api_key or not notion_db_id:
+    raise ValueError("Notion API key or database ID is not set.")
+
+client = Groq(api_key=api_key)
+
+# Helper Functions for Matching Game
 def parse_csv(file):
     global remaining_pairs
     try:
@@ -59,6 +79,57 @@ def reset_game():
     remaining_pairs = 0
     return "Game reset. Upload a new file or add words to start again.", vocabulary
 
+# New Helper Functions for Groq Chat Integration
+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}}]}
+        }
+    }
+    response = requests.post(url, headers=headers, json=data)
+    if response.status_code != 200:
+        print(f"Failed to log to Notion: {response.text}")
+
+def chat_with_groq(input_text, chat_history):
+    try:
+        messages = [
+            {"role": "system", "content": (
+                "You are an English dictionary and quiz designer tutor. When you are given a vocabulary, do two things:\n"
+                "(1) Provide a definition within 50 words\n(2) Provide a sample sentence under 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})
+        completion = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=messages,
+            temperature=1,
+            max_tokens=1024
+        )
+        response = completion.choices[0].message.content
+        return response
+    except Exception as e:
+        return f"Error occurred: {e}"
+
+def respond(name, message, chat_history):
+    response = chat_with_groq(message, chat_history)
+    chat_history.append((message, response))
+    log_to_notion(name, message, response)
+    return chat_history, ""
+
 # Gradio Interface
 def main():
     with gr.Blocks() as app:
@@ -79,11 +150,18 @@ def main():
             score_output = gr.Textbox(label="Score")
             remaining_output = gr.Textbox(label="Remaining Pairs")
 
+        with gr.Tab("English Vocab Bot"):
+            chatbot = gr.Chatbot()
+            name_input = gr.Textbox(label="Name", placeholder="Enter your name")
+            message_input = gr.Textbox(placeholder="Input your vocab...")
+            send_button = gr.Button("Submit")
+            chat_history = gr.State([])
+            send_button.click(respond, [name_input, message_input, chat_history], [chatbot, message_input])
+
         with gr.Tab("Reset Game"):
             reset_button = gr.Button("Reset")
             reset_output = gr.Textbox(label="Reset Status")
 
-        # Actions
         upload_button.click(parse_csv, inputs=[csv_upload], outputs=[upload_output])
         start_button.click(start_game, outputs=[game_status, word_dropdown, definition_dropdown])
         match_button.click(check_match, inputs=[word_dropdown, definition_dropdown], outputs=[match_output, score_output, remaining_output])
@@ -91,7 +169,6 @@ def main():
 
     return app
 
-# Run Application
 if __name__ == "__main__":
     app = main()
     app.launch()