Update app.py

app.py

@@ -6,6 +6,9 @@ import os
 import shutil
 from datetime import datetime
 import Arcana
+from nylon import *
+import pandas as pd
+import json
 
 # SSL configuration to avoid verification issues
 try:
@@ -15,6 +18,24 @@ except AttributeError:
 else:
     ssl._create_default_https_context = _create_unverified_https_context
 
+def query_database2(query):
+    db = ChatDatabase('memory.txt')
+
+    sender = 'Arcana'
+    N = 10
+    cache = {}
+    query_tag = None
+
+    relevant_messages = db.get_relevant_messages(sender, query, N, cache, query_tag)
+
+    print("Relevant messages:")
+    for message in relevant_messages:
+        print(f"Sender: {message[0]}, Time: {message[1]}, Tag: {message[3]}")
+        print(f"Message: {message[2][:100]}...")
+        print()
+
+    df_data = [str(message) for message in relevant_messages]
+    return ';'.join(df_data)
 
 # OpenAI client setup
 client = OpenAI(
@@ -22,20 +43,71 @@ client = OpenAI(
     api_key='sk-Nxf8HmLpfIMhCd83n3TOr00TR57uBZ0jMbAgGCOzppXvlsx1',
 )
 
+# Function list for OpenAI API
+function_list = [
+    {
+        "name": "query_database",
+        "description": "Query the database and return a list of results as strings",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "query": {
+                    "type": "string",
+                    "description": "The query to execute against the database"
+                },
+            },
+            "required": ["query"]
+        }
+    }
+]
+
+# Mapping of function names to actual function objects
+function_map = {
+    "query_database": query_database2
+}
+
+def execute_function(function_name, function_args):
+    if function_name in function_map:
+        return function_map[function_name](**function_args)
+    else:
+        return f"Error: Function {function_name} not found"
+
 # Retry logic for OpenAI API call
 def openai_api_call(messages, retries=3, delay=5):
     for attempt in range(retries):
         try:
             completion = client.chat.completions.create(
-                model="gpt-4o",
+                model="gpt-3.5-turbo",  # Changed from "gpt-4o" to "gpt-4"
                 messages=messages,
+                functions=function_list,
+                function_call='auto',
                 timeout=10
             )
-            return completion.choices[0].message.content
+            response_message = completion.choices[0].message
+
+            # Check if the model wants to call a function
+            if response_message.function_call:
+                function_name = response_message.function_call.name
+                function_args = json.loads(response_message.function_call.arguments)
+                function_response = execute_function(function_name, function_args)
+                # Add the function response to the conversation
+                messages.append(response_message.model_dump())  # The model's request to call the function
+                messages.append({
+                    "role": "function",
+                    "name": function_name,
+                    "content": json.dumps(function_response)
+                })
+                # Make a follow-up call to the model with the function response
+                return openai_api_call(messages)
+            else:
+                return response_message.content
+
         except Exception as e:
             print(f"Attempt {attempt + 1} failed: {e}")
-            time.sleep(delay)
-    return "Sorry, I am having trouble connecting to the server. Please try again later."
+            if attempt < retries - 1:
+                time.sleep(delay)
+            else:
+                return "Sorry, I am having trouble connecting to the server. Please try again later."
 
 # Chatbot response function
 def chatbot_response(message, history):
@@ -123,7 +195,9 @@ def display_file(evt: gr.SelectData, df):
 def render_to_database():
     # This function is undefined as per your request
     Arcana.main()
-    
+
+def change_theme(theme):
+    gr.Interface.theme = theme
 
 def rename_file(new_name):
     global selected
@@ -138,28 +212,75 @@ def rename_file(new_name):
             return list_uploaded_files(), f"File {selected} not found", None, "", ""
     return list_uploaded_files(), "No file selected or new name not provided", None, "", ""
 
+def query_database(query):
+    # Usage example
+    db = ChatDatabase('memory.txt')
+
+    # Example 1: Get relevant messages
+    sender = 'Arcana'
+    N = 10
+    cache = {}
+    query_tag = None
+
+    relevant_messages = db.get_relevant_messages(sender, query, N, cache, query_tag)
+
+    print("Relevant messages:")
+    for message in relevant_messages:
+        print(f"Sender: {message[0]}, Time: {message[1]}, Tag: {message[3]}")
+        print(f"Message: {message[2][:100]}...")
+        print()
+
+    df_data = [{"Nylon Returned Query": str(message)} for message in relevant_messages]
+    
+    # Create a pandas DataFrame
+    df = pd.DataFrame(df_data)
+    
+    return df
+
+    
+example_database = [
+    "What is Hydrogen Bonding?",
+    "Tell me the difference between impulse and force.",
+    "Tell me a joke that Calculus students will understand.",
+    "How should I review for the AP Biology Exam?",
+    "What kind of resources are available in PA and Indexademics?",
+    "What is the StandardCAS™ group?",
+    "Explain the concept of quantum entanglement.",
+    "What are the main differences between mitosis and meiosis?",
+    "How does the Doppler effect work?",
+    "Explain the process of photosynthesis.",
+    "What is the significance of the Pythagorean theorem?",
+    "How does natural selection contribute to evolution?",
+    "What is the most important chapter in AP Statistics?",
+    "How should I prepare on the IB Chinese Exam?"
+]
+
+import random
+
+def get_random_examples(num_examples=5):
+    return random.sample(example_database, min(num_examples, len(example_database)))
+
 # Create the Gradio interface for the chatbot
 chatbot_interface = gr.ChatInterface(
     chatbot_response,
     chatbot=gr.Chatbot(height=400),
-    textbox=gr.Textbox(placeholder="Type your message here...", container=True, scale=100),
+    textbox=gr.Textbox(placeholder="Type your message here...", container=True, scale=10),
     title="Review With Arcana",
     description="ArcanaUI v0.8 - Chatbot",
     theme="soft",
-    examples=[
-        "What is Hydrogen Bonding?",
-        "Tell me the difference between impulse and force.",
-        "Tell me a joke that Calculus students will know.",
-        "How should I review for the AP Biology Exam?",
-        "What kind of resources are available in PA and Indexademics?",
-        "What is the StandardCAS™ group?"
-    ],
+    examples=get_random_examples(),
     cache_examples=False,
     retry_btn=None,
     undo_btn="Delete Previous",
     clear_btn="Clear"
 )
 
+
+def relaunch():
+    global demo
+    demo.close()
+    demo.launch(share=True)
+    
 # Combine the interfaces using Tabs
 with gr.Blocks() as demo:
     gr.Markdown("# ArcanaUI v0.8")
@@ -215,8 +336,29 @@ with gr.Blocks() as demo:
             render_button = gr.Button("Render all PDFs to Database")
             render_button.click(fn=render_to_database)
 
-        with gr.Tabitem('Settings'):
-            gr.Markdown('Settings')
+        with gr.TabItem('Settings'):
+            with gr.TabItem('Database'):
+                gr.Markdown('Settings')
+
+                test_nylon = gr.Textbox(label='Test Nylon', placeholder='Query')
+                uploaded_files_list2 = gr.DataFrame(headers=["Nylon Returned Query"], datatype="str", interactive=False)
+                
+                query_button = gr.Button('Query')
+                
+                query_button.click(fn=query_database, inputs=test_nylon, outputs=uploaded_files_list2)
+            with gr.TabItem('Theme'):
+                gr.Markdown('Change Theme')
+
+                theme_dropdown = gr.Dropdown(choices=['default', 'compact', 'huggingface', 'soft', 'dark'], label='Choose Theme')
+                theme_button = gr.Button('Apply Theme')
+
+                theme_button.click(fn=change_theme, inputs=theme_dropdown)
+                relaunch_button = gr.Button('Relaunch')
+                relaunch_button.click(fn=relaunch)
+
 
 # Launch the interface
 demo.launch(share=True)
+
+
+