basic app

app.py

@@ -1,30 +1,128 @@
 import gradio as gr
+import os
 from openai import OpenAI
 from utils import get_secret
+from database import create_chatbot, get_chatbot, update_chatbot, delete_chatbot, get_all_chatbots, filter_profanity
 
-# Initialize the OpenAI client
-client = OpenAI(api_key=get_secret('OPENAI_API_KEY'))
+CREATE_APP_PW = get_secret("CREATE_APP_PW")
+ADMIN_PW = get_secret("ADMIN_PW")
+OPENAI_API_KEY = get_secret("OPENAI_API_KEY")
+
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+
+def create_chatbot_interface(name, custom_instruction, password):
+    if password != CREATE_APP_PW:
+        return "Invalid password. Chatbot creation failed."
+
+    filtered_name = filter_profanity(name)
+    filtered_instruction = filter_profanity(custom_instruction)
+
+    chatbot = create_chatbot(filtered_name, filtered_instruction)
+    return f"Chatbot created with ID: {chatbot.chatbot_id}"
+
+def chat_with_bot(message, history, chatbot_id):
+    chatbot = get_chatbot(chatbot_id)
+    if not chatbot:
+        return "Invalid chatbot ID or chatbot not active"
+
+    # Prepare the conversation history for OpenAI
+    conversation = [
+        {"role": "system", "content": chatbot.custom_instruction},
+    ]
+    for human, assistant in history:
+        conversation.append({"role": "user", "content": human})
+        conversation.append({"role": "assistant", "content": assistant})
+    conversation.append({"role": "user", "content": message})
 
-def generate_response(prompt):
     try:
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": prompt}
-            ]
-        )
+        response = client.chat.completions.create(model="gpt-3.5-turbo",
+        messages=conversation)
         return response.choices[0].message.content
     except Exception as e:
-        return str(e)
-
-demo = gr.Interface(
-    fn=generate_response,
-    inputs="text",
-    outputs="text",
-    title="OpenAI GPT-3.5 Turbo Chatbot",
-    description="Enter a prompt to get a response from OpenAI's GPT-3.5 Turbo model."
-)
+        return f"An error occurred: {str(e)}"
+
+def admin_view(password):
+    if password != ADMIN_PW:
+        return "Invalid admin password."
+
+    chatbots = get_all_chatbots()
+    return gr.Dataframe(
+        headers=["ID", "Name", "Custom Instruction", "Is Active"],
+        data=[[c.chatbot_id, c.name, c.custom_instruction, c.is_active] for c in chatbots]
+    )
+
+def admin_action(action, chatbot_id, name, custom_instruction, is_active, password):
+    if password != ADMIN_PW:
+        return "Invalid admin password."
+
+    if action == "Edit":
+        update_chatbot(chatbot_id, name, custom_instruction, is_active)
+        return "Chatbot updated successfully."
+    elif action == "Delete":
+        delete_chatbot(chatbot_id)
+        return "Chatbot deleted successfully."
+    else:
+        return "Invalid action."
+
+with gr.Blocks() as demo:
+    chatbot_id_input = gr.Textbox(label="Enter Chatbot ID", visible=False)
+
+    with gr.Tab("Chat"):
+        chatbot_title = gr.Markdown("Welcome to the Chatbot")
+        chat_interface = gr.ChatInterface(
+            chat_with_bot,
+            additional_inputs=[chatbot_id_input],
+        )
+        share_link = gr.Textbox(label="Share Link", interactive=False)
+
+    with gr.Tab("Create Chatbot"):
+        name = gr.Textbox(label="Chatbot Name")
+        instructions = gr.Textbox(label="Enter custom instructions for your chatbot")
+        create_password = gr.Textbox(label="Creation Password", type="password")
+        create_button = gr.Button("Create Chatbot")
+        create_output = gr.Textbox(label="Creation Result")
+        create_button.click(create_chatbot_interface, inputs=[name, instructions, create_password], outputs=create_output)
+
+    with gr.Tab("Admin"):
+        admin_password = gr.Textbox(label="Admin Password", type="password")
+        admin_button = gr.Button("View Chatbots")
+        admin_output = gr.Dataframe(headers=["ID", "Name", "Custom Instruction", "Is Active"])
+        admin_button.click(admin_view, inputs=[admin_password], outputs=admin_output)
+
+        with gr.Row():
+            admin_action_dropdown = gr.Dropdown(["Edit", "Delete"], label="Action")
+            admin_chatbot_id = gr.Textbox(label="Chatbot ID")
+            admin_name = gr.Textbox(label="Name")
+            admin_instruction = gr.Textbox(label="Custom Instruction")
+            admin_is_active = gr.Checkbox(label="Is Active")
+        admin_action_button = gr.Button("Perform Action")
+        admin_action_output = gr.Textbox(label="Action Result")
+        admin_action_button.click(
+            admin_action,
+            inputs=[admin_action_dropdown, admin_chatbot_id, admin_name, admin_instruction, admin_is_active, admin_password],
+            outputs=admin_action_output
+        )
+
+    @demo.load(inputs=[chatbot_id_input], outputs=[chatbot_title, chatbot_id_input, share_link])
+    def load_chatbot(chatbot_id, request: gr.Request):
+        # Check if there's a chatbot_id in the URL parameters
+        params = request.query_params
+        if "chatbot_id" in params:
+            chatbot_id = params["chatbot_id"]
+
+        chatbot = get_chatbot(chatbot_id)
+        if chatbot:
+            return (
+                gr.update(value=f"Welcome to {chatbot.name}"),
+                gr.update(value=chatbot_id, visible=True),
+                gr.update(value=f"https://huggingface.co/spaces/codora/ai-app-creator?chatbot_id={chatbot_id}")
+            )
+        return (
+            gr.update(value="Welcome to the Chatbot"),
+            gr.update(value="", visible=True),
+            gr.update(value="")
+        )
 
 if __name__ == "__main__":
     demo.launch()