Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import api | |
| import utils | |
| import os | |
| import shutil | |
| import json | |
| import logging | |
| from transformers import pipeline | |
| # Setup logging | |
| logging.basicConfig(level=logging.INFO) | |
| logger = logging.getLogger(__name__) | |
| FILE_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| EXAMPLES_PATH = os.path.join(FILE_DIR, 'examples.json') | |
| OUTPUT_DIR = os.path.join(os.path.dirname(FILE_DIR), "auto_gpt_workspace") | |
| if not os.path.exists(OUTPUT_DIR): | |
| os.mkdir(OUTPUT_DIR) | |
| CSS = """ | |
| #chatbot {font-family: monospace;} | |
| #files .generating {display: none;} | |
| #files .min {min-height: 0px;} | |
| """ | |
| def get_api_key(): | |
| return gr.Textbox(label="Hugging Face API Key", type="password") | |
| def get_ai_name(): | |
| return gr.Textbox(label="AI Name", placeholder="e.g. Entrepreneur-GPT") | |
| def get_ai_role(): | |
| return gr.Textbox(label="AI Role", placeholder="e.g. an AI designed to autonomously develop and run businesses with the sole goal of increasing your net worth.") | |
| def get_description(): | |
| return gr.Textbox(label="Description", placeholder="Enter a brief description of the project.") | |
| def get_top_5_goals(): | |
| return gr.Dataframe(row_count=(5, "fixed"), col_count=(1, "fixed"), headers=["AI Goals - Enter up to 5"], type="array") | |
| def get_example_values(): | |
| try: | |
| with open(EXAMPLES_PATH, 'r', encoding='utf-8') as f: | |
| return json.load(f) | |
| except Exception as e: | |
| logger.error(f"Error loading examples: {e}") | |
| return [] | |
| def get_chatbot(): | |
| return gr.Chatbot(elem_id="chatbot", type='messages') | |
| def get_yes_btn(): | |
| return gr.Button("Yes", variant="primary", interactive=False) | |
| def get_consecutive_yes(): | |
| return gr.Slider(1, 10, 1, step=1, label="Consecutive Yes", interactive=False) | |
| def get_custom_response(): | |
| return gr.Textbox(label="Custom Response", placeholder="Press 'Enter' to Submit.", interactive=False) | |
| def get_progress(): | |
| return gr.Progress() | |
| def get_generated_files(): | |
| return gr.HTML(lambda: f"Generated Files<pre><code style='overflow-x: auto'>{utils.format_directory(OUTPUT_DIR)}</pre></code>", every=3, elem_id="files") | |
| def get_download_btn(): | |
| return gr.Button("Download All Files") | |
| def get_inferred_tasks(): | |
| return gr.Textbox(label="Inferred Tasks", interactive=False) | |
| def start(huggingface_key, ai_name, ai_role, top_5_goals, description): | |
| try: | |
| from api import AutoAPI | |
| auto_api = AutoAPI(huggingface_key, ai_name, ai_role, top_5_goals) | |
| logger.info("AutoAPI started with AI Name: %s, AI Role: %s", ai_name, ai_role) | |
| # Infer tasks based on the role and description | |
| tasks = auto_api.infer_tasks(description) | |
| logger.info("Inferred tasks: % s", tasks) | |
| return gr.Column.update(visible=False), gr.Column.update(visible=True), auto_api, tasks | |
| except Exception as e: | |
| logger.error("Failed to start AutoAPI: %s", str(e)) | |
| return gr.Column.update(visible=True), gr.Column.update(visible=False), None, [] | |
| # Main Gradio Interface | |
| with gr.Blocks(css=CSS) as demo: | |
| gr.Markdown("# AutoGPT Task Inference") | |
| with gr.Row(): | |
| api_key = get_api_key() | |
| ai_name = get_ai_name() | |
| ai_role = get_ai_role() | |
| description = get_description() | |
| top_5_goals = get_top_5_goals() | |
| start_btn = gr.Button("Start") | |
| main_pane = gr.Column(visible=False) | |
| setup_pane = gr.Column(visible=True) | |
| inferred_tasks = get_inferred_tasks() | |
| start_btn.click( | |
| start, | |
| [api_key, ai_name, ai_role, top_5_goals, description], | |
| [setup_pane, main_pane, None, inferred_tasks], | |
| ) | |
| with main_pane: | |
| get_generated_files() | |
| get_download_btn() | |
| # Launch the Gradio app | |
| if __name__ == "__main__": | |
| demo.launch() |