test

babyagi/babyagi.py

@@ -25,6 +25,84 @@ import torch
 import numpy
 
 import psycopg2
+import shutil
+import gradio as gr
+from mysite.libs.utilities import chat_with_interpreter, completion, process_file
+from interpreter import interpreter
+import mysite.interpreter.interpreter_config  # インポートするだけで設定が適用されます
+import importlib
+import os
+import pkgutil
+import async_timeout
+import asyncio
+
+
+DESCRIPTION = """
+<div>
+<h1 style="text-align: center;">develop site</h1>
+<p>🦕 共同開発 AIシステム設定 LINE開発 CHATGPTS CHATGPTアシスタント設定 AI自動開発設定 APPSHEET GAS PYTHON</p>
+</div>
+<!-- Start of HubSpot Embed Code -->
+  <script type="text/javascript" id="hs-script-loader" async defer src="//js-na1.hs-scripts.com/46277896.js"></script>
+<!-- End of HubSpot Embed Code -->
+"""
+
+LICENSE = """
+<p/>
+<!-- Start of HubSpot Embed Code -->
+  <script type="text/javascript" id="hs-script-loader" async defer src="//js-na1.hs-scripts.com/46277896.js"></script>
+<!-- End of HubSpot Embed Code -->
+---
+Built with Meta Llama 3
+"""
+
+PLACEHOLDER = """
+<div style="padding: 30px; text-align: center; display: flex; flex-direction: column; align-items: center;">
+   <img src="https://ysharma-dummy-chat-app.hf.space/file=/tmp/gradio/8e75e61cc9bab22b7ce3dec85ab0e6db1da5d107/Meta_lockup_positive%20primary_RGB.jpg" style="width: 80%; max-width: 550px; height: auto; opacity: 0.55;  ">
+   <h1 style="font-size: 28px; margin-bottom: 2px; opacity: 0.55;">Meta llama3</h1>
+   <p style="font-size: 18px; margin-bottom: 2px; opacity: 0.65;">Ask me anything...</p>
+</div>
+"""
+
+
+# チャットインターフェースの関数定義
+# def chat_with_interpreter(message):
+#    return "Response: " + message
+
+
+# カスタムCSSの定義
+css = """
+.gradio-container {
+    height: 100vh; /* 全体の高さを100vhに設定 */
+    display: flex;
+    flex-direction: column;
+}
+.gradio-tabs {
+    flex: 1; /* タブ全体の高さを最大に設定 */
+    display: flex;
+    flex-direction: column;
+}
+.gradio-tab-item {
+    flex: 1; /* 各タブの高さを最大に設定 */
+    display: flex;
+    flex-direction: column;
+    overflow: hidden; /* オーバーフローを隠す */
+}
+.gradio-block {
+    flex: 1; /* ブロックの高さを最大に設定 */
+    display: flex;
+    flex-direction: column;
+}
+.gradio-chatbot {
+    height: 100vh; /* チャットボットの高さを100vhに設定 */
+    overflow-y: auto; /* 縦スクロールを有効にする */
+}
+"""
+GENERATION_TIMEOUT_SEC = 60
+# Gradio block
+chatbot2 = gr.Chatbot(height=450, placeholder=PLACEHOLDER, label="Gradio ChatInterface")
+
+
 
 class ProductDatabase:
     def __init__(self, database_url):
@@ -771,6 +849,114 @@ def main():
 if __name__ == "__main__":
     main()
 
+
+async def completion(message: str, history, c=None, d=None, prompt="あなたは日本語の優秀なアシスタントです。"):
+    OBJECTIVE = message
+    loop = True
+    while loop:
+        # As long as there are tasks in the storage...
+        if not tasks_storage.is_empty():
+        #OBJECTIVE = "ボットの性能をよくする方法　日本語で説明"
+            # Print the task list
+            print("\033[95m\033[1m" + "\n*****TASK LIST*****\n" + "\033[0m\033[0m")
+            for t in tasks_storage.get_task_names():
+                print(" • " + str(t))
+                result yield
+
+            # Step 1: Pull the first incomplete task
+            task = tasks_storage.popleft()
+            print("\033[92m\033[1m" + "\n*****NEXT TASK*****\n" + "\033[0m\033[0m")
+            print(str(task["task_name"]))
+            yield result
+
+            # Send to execution function to complete the task based on the context
+            result = execution_agent(OBJECTIVE, str(task["task_name"]))
+            print("\033[93m\033[1m" + "\n*****TASK RESULT*****\n" + "\033[0m\033[0m")
+            print(result)
+            yield result
+
+            # Step 2: Enrich result and store in the results storage
+            # This is where you should enrich the result if needed
+            enriched_result = {
+                "data": result
+            }
+            # extract the actual result from the dictionary
+            # since we don't do enrichment currently
+            vector = enriched_result["data"]
+
+            result_id = f"result_{task['task_id']}"
+
+            results_storage.add(task, result, result_id)
+
+            # Step 3: Create new tasks and re-prioritize task list
+            # only the main instance in cooperative mode does that
+            new_tasks = task_creation_agent(
+                OBJECTIVE,
+                enriched_result,
+                task["task_name"],
+                tasks_storage.get_task_names(),
+            )
+
+            print('Adding new tasks to task_storage')
+            for new_task in new_tasks:
+                new_task.update({"task_id": tasks_storage.next_task_id()})
+                print(str(new_task))
+                tasks_storage.append(new_task)
+
+            if not JOIN_EXISTING_OBJECTIVE:
+                prioritized_tasks = prioritization_agent()
+                if prioritized_tasks:
+                    tasks_storage.replace(prioritized_tasks)
+
+            # Sleep a bit before checking the task list again
+            time.sleep(15)
+        else:
+            print('Done.')
+            loop = False
+
+
+
+
+with gr.Blocks(fill_height=True, css=css) as gradio_babyagi:
+    # gr.Markdown(DESCRIPTION)
+    # gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
+    gr.ChatInterface(
+        fn=completion,
+        chatbot=chatbot2,
+        fill_height=True,
+        additional_inputs_accordion=gr.Accordion(
+            label="⚙️ Parameters", open=False, render=False
+        ),
+        additional_inputs=[
+            gr.Slider(
+                minimum=0,
+                maximum=1,
+                step=0.1,
+                value=0.95,
+                label="Temperature",
+                render=False,
+            ),
+            gr.Slider(
+                minimum=128,
+                maximum=4096,
+                step=1,
+                value=512,
+                label="Max new tokens",
+                render=False,
+            ),
+        ],
+        examples=[
+            ["HTMLのサンプルを作成して"],
+            [
+                "CUDA_VISIBLE_DEVICES=0 llamafactory-cli train examples/lora_single_gpu/llama3_lora_sft.yaml"
+            ],
+        ],
+        cache_examples=False,
+    )
+
+    gr.Markdown(LICENSE)
+   
+
 def test_postgres():
     # データベース接続情報
     DATABASE_URL = "postgresql://miyataken999:yz1wPf4KrWTm@ep-odd-mode-93794521.us-east-2.aws.neon.tech/neondb?sslmode=require"

mysite/routers/gradio.py

@@ -39,9 +39,9 @@ def include_gradio_interfaces():
 
 def setup_gradio_interfaces():
 
-
-    default_interfaces = []#,demo]
-    default_names = ["Chat","OpenInterpreter","DataBase","CreateFromDOC"]#"demo"]
+    from babyagi.babyagi import gradio_babyagi
+    default_interfaces = [gradio_babyagi]#,demo]
+    default_names = ["Chat","OpenInterpreter","DataBase","CreateFromDOC","babyagi"]#"demo"]
 
     gradio_interfaces, gradio_names = include_gradio_interfaces()