Added API

blubeAPI.py

@@ -1,5 +1,18 @@
 import gradio as gr
 import openai
+import json
+import time
+import sys
+from gpt_index import SimpleDirectoryReader, GPTListIndex, GPTSimpleVectorIndex, LLMPredictor, PromptHelper
+from langchain.chat_models import ChatOpenAI
+import gradio as gr
+import openai
+from openai import OpenAI
+import os
+
+os.environ["OPENAI_API_KEY"] = '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'
+
+client = OpenAI()
 
 def format(data: str):
     data = data.replace("['", "")
@@ -9,7 +22,7 @@ def format(data: str):
     data = data.replace("\\\'", "\'")
     data = data.replace("\\\n", "")
     return data
-def runbot(chatbot, client: openai.OpenAI):
+def runbot():
     iface = gr.Interface(fn=chatbot,
                         inputs=gr.components.Textbox(lines=7, label="Enter your text"),
                         outputs="text",
@@ -51,4 +64,90 @@ def getAssistant():
 def getPersona():
     return persona
 def getThread():
-    return thread
+    return thread
+def chatbot(input_text):
+    my_assistant = getAssistant()
+    persona = getPersona()
+
+    print("(1/4) Message sent, creating thread...\n")
+    # Create a thread for the assistant
+    thread = client.beta.threads.create()
+    message = client.beta.threads.messages.create(
+        thread_id=thread.id,
+        role="user",
+        content=f"{input_text}"
+    )
+
+    print("(2/4) Thread created, running thread...\n")
+
+    # Run the thread
+    run = client.beta.threads.runs.create(
+        thread_id=thread.id,
+        assistant_id=my_assistant.id,
+        instructions=f"{persona}"
+    )
+    print("(3/4) Thread active, getting thread...\n")
+
+    # Get the thread
+    run = client.beta.threads.runs.retrieve(
+        thread_id=thread.id,
+        run_id=run.id
+    )
+    print("(4/4) Thread received, awaiting chat completion\n")
+
+    # Wait for completion
+    t: int = 0
+    tt = 0
+    errorCode = 0
+    while run.status != "completed":
+        run = client.beta.threads.runs.retrieve(
+            thread_id=thread.id, run_id=run.id
+        )
+        s = run.status
+
+        if s == "failed":
+            print(f"Process failed after {tt} ticks at {run.failed_at} with error stack:\n{run.last_error}\n\n")
+            errorCode = 1
+            break
+        elif s == "cancelled":
+            print(f"Process cancelled after {tt} ticks at {run.cancelled_at} with error stack:\n{run.last_error}\n\n")
+            errorCode = 2
+        elif s == "in_progress" or s == "queued":
+            t = t + 1
+            tt = tt + 1
+        else:
+            print(s)
+        if t > 15:
+            print(f"Time taken is over 15 ticks, average time is 11, check issues.\n\nINFO: Status is {s} and the total ticks is {tt}")
+            t = 0
+    if errorCode == 1:
+        raise RuntimeError("\n\nThe process finished with an error: Process failed unexpectedly. Check the command prompt for more details. This error is most likely a overuse error, so the program will sleep for 20 seconds to avoid more errors\n\n")
+        time.sleep(20)
+        print("\n20 seconds is over: requests are up")
+    elif errorCode == 2:
+        raise RuntimeError("The process finished with an error: Process cancelled unexpectedly. Check the command prompt for more details.")
+    else:
+        messages = client.beta.threads.messages.list(
+            thread_id=thread.id
+        )    
+        messages = messages.json()
+        print(f"Process completed in {tt} ticks at {run.completed_at}")
+        
+        mesf = open("message.json", "w")
+        mesf.write(messages)
+        mesf.close()
+
+        mesf = open("./message.json", "rt")
+        mess = json.load(mesf)
+        mess = mess["data"]
+
+        all_msgs = []
+        l = 0
+        for m in mess:
+            l = l + 1
+        for i in range(l - 1):
+            all_msgs.append(mess[i]["content"][0]["text"]["value"])
+
+        all_msgs = str(all_msgs)
+        all_msgs = format(all_msgs)
+        return all_msgs