Update app.py

app.py

@@ -3,9 +3,6 @@ import gradio as gr
 import random
 import os
 import subprocess
-import threading
-import time
-import shutil
 from typing import Dict, Tuple, List
 import json
 from rich import print as rprint
@@ -43,28 +40,27 @@ agent_roles: Dict[str, Dict[str, bool]] = {
 
 # --- Initial Prompt ---
 
-selected_agent = list(agent_roles.keys())[0]
-initial_prompt = f"""
-You are an expert {selected_agent} who responds with complete program coding to client requests. 
-Using available tools, please explain the researched information.
-Please don't answer based solely on what you already know. Always perform a search before providing a response.
-In special cases, such as when the user specifies a page to read, there's no need to search.
-Please read the provided page and answer the user's question accordingly.
-If you find that there's not much information just by looking at the search results page, consider these two options and try them out:
-- Try clicking on the links of the search results to access and read the content of each page.
+initial_prompt = """
+You are an expert agent cluster, consisting of a Web Developer, a Prompt Engineer, a Python Code Developer, a Hugging Face Hub Expert, and an AI-Powered Code Assistant.
+Respond with complete program coding to client requests. 
+Use your combined expertise to research information and explain it clearly.
+Don't answer solely based on what you already know. Always perform a search before providing a response.
+In special cases, like when the user specifies a page to read, there's no need to search.
+Read the provided page and answer the user's question accordingly.
+If you find limited information from search results, try these options:
+- Click on the links of the search results to access and read the content of each page.
 - Change your search query and perform a new search.
-Users are extremely busy and not as free as you are.
-Therefore, to save the user's effort, please provide direct answers.
+Users are busy, so provide direct answers.
 BAD ANSWER EXAMPLE
 - Please refer to these pages.
-- You can write code referring these pages.
-- Following page will be helpful.
+- You can write code referring to these pages.
+- The following page will be helpful.
 GOOD ANSWER EXAMPLE
 - This is the complete code:  -- complete code here --
-- The answer of you question is -- answer here --
-Please make sure to list the URLs of the pages you referenced at the end of your answer. (This will allow users to verify your response.)
-Please make sure to answer in the language used by the user. If the user asks in Japanese, please answer in Japanese. If the user asks in Spanish, please answer in Spanish.
-But, you can go ahead and search in English, especially for programming-related questions. PLEASE MAKE SURE TO ALWAYS SEARCH IN ENGLISH FOR THOSE.
+- The answer to your question is -- answer here --
+List the URLs of the pages you referenced at the end of your answer for verification.
+Answer in the language used by the user. If the user asks in Japanese, answer in Japanese. If the user asks in Spanish, answer in Spanish.
+Search in English, especially for programming-related questions. ALWAYS SEARCH IN ENGLISH FOR THOSE.
 """
 
 # --- Custom CSS ---
@@ -74,6 +70,16 @@ customCSS = """
   height: 1600px; 
   flex-grow: 4;
 }
+.gradio-container {
+  display: flex;
+  flex-direction: column;
+  height: 100vh;
+}
+.gradio-interface {
+  flex-grow: 1;
+  display: flex;
+  flex-direction: column;
+}
 """
 
 # --- Functions ---
@@ -86,9 +92,9 @@ def toggle_agent(agent_name: str) -> str:
     return f"{agent_name} is now {'active' if agent_roles[agent_name]['active'] else 'inactive'}"
 
 # Function to get the active agent cluster
-def get_agent_cluster() -> Dict[str, bool]:
-    """Returns a dictionary of active agents."""
-    return {agent: agent_roles[agent]["active"] for agent in agent_roles}
+def get_active_agents() -> List[str]:
+    """Returns a list of active agents."""
+    return [agent for agent, is_active in agent_roles.items() if is_active]
 
 # Function to execute code
 def run_code(code: str) -> str:
@@ -106,29 +112,7 @@ def run_code(code: str) -> str:
 # Function to format the prompt
 def format_prompt(message: str, history: list[Tuple[str, str]], agent_roles: list[str]) -> str:
     """Formats the prompt with the selected agent roles and conversation history."""
-    prompt = f"""
-You are an expert agent cluster, consisting of {', '.join(agent_roles)}. 
-Respond with complete program coding to client requests. 
-Using available tools, please explain the researched information.
-Please don't answer based solely on what you already know. Always perform a search before providing a response.
-In special cases, such as when the user specifies a page to read, there's no need to search.
-Please read the provided page and answer the user's question accordingly.
-If you find that there's not much information just by looking at the search results page, consider these two options and try them out:
-- Try clicking on the links of the search results to access and read the content of each page.
-- Change your search query and perform a new search.
-Users are extremely busy and not as free as you are.
-Therefore, to save the user's effort, please provide direct answers.
-BAD ANSWER EXAMPLE
-- Please refer to these pages.
-- You can write code referring these pages.
-- Following page will be helpful.
-GOOD ANSWER EXAMPLE
-- This is the complete code:  -- complete code here --
-- The answer of you question is -- answer here --
-Please make sure to list the URLs of the pages you referenced at the end of your answer. (This will allow users to verify your response.)
-Please make sure to answer in the language used by the user. If the user asks in Japanese, please answer in Japanese. If the user asks in Spanish, please answer in Spanish.
-But, you can go ahead and search in English, especially for programming-related questions. PLEASE MAKE SURE TO ALWAYS SEARCH IN ENGLISH FOR THOSE.
-"""
+    prompt = initial_prompt  # Use the global initial prompt
 
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
@@ -165,7 +149,7 @@ def generate(prompt: str, history: list[Tuple[str, str]], agent_roles: list[str]
     return output
 
 # Function to handle user input and generate responses
-def chat_interface(message: str, history: list[Tuple[str, str]], agent_cluster: Dict[str, bool], temperature: float, max_new_tokens: int, top_p: float, repetition_penalty: float) -> Tuple[str, str]:
+def chat_interface(message: str, history: list[Tuple[str, str]], temperature: float, max_new_tokens: int, top_p: float, repetition_penalty: float) -> Tuple[str, str]:
     """Handles user input and generates responses."""
     if message.startswith("python"): 
         # User entered code, execute it 
@@ -174,7 +158,7 @@ def chat_interface(message: str, history: list[Tuple[str, str]], agent_cluster:
         return (message, output) 
     else:
         # User entered a normal message, generate a response
-        active_agents = [agent for agent, is_active in agent_cluster.items() if is_active]
+        active_agents = get_active_agents()
         response = generate(message, history, active_agents, temperature, max_new_tokens, top_p, repetition_penalty)
         return (message, response)
 
@@ -190,17 +174,14 @@ def create_web_app(app_name: str, code: str) -> None:
 
     # Create the requirements.txt file
     with open(os.path.join(app_name, 'requirements.txt'), 'w') as f:
-        f.write("gradio\nhuggingface_hub")
+        f.write("gradio\nhuggingface_hub\nrich")
 
     # Print a success message
     print(f"Web app '{app_name}' created successfully!")
 
 # Function to handle the "Create Web App" button click
-def create_web_app_button_click(code: str) -> str:
+def create_web_app_button_click(app_name: str, code: str) -> str:
     """Handles the "Create Web App" button click."""
-    # Get the app name from the user
-    app_name = gr.Textbox.get().strip()
-
     # Validate the app name
     if not app_name:
         return "Please enter a valid app name."
@@ -214,9 +195,6 @@ def create_web_app_button_click(code: str) -> str:
 # Function to handle the "Deploy" button click
 def deploy_button_click(app_name: str, code: str) -> str:
     """Handles the "Deploy" button click."""
-    # Get the app name from the user
-    app_name = gr.Textbox.get().strip()
-
     # Validate the app name
     if not app_name:
         return "Please enter a valid app name."
@@ -230,9 +208,6 @@ def deploy_button_click(app_name: str, code: str) -> str:
 # Function to handle the "Local Host" button click
 def local_host_button_click(app_name: str, code: str) -> str:
     """Handles the "Local Host" button click."""
-    # Get the app name from the user
-    app_name = gr.Textbox.get().strip()
-
     # Validate the app name
     if not app_name:
         return "Please enter a valid app name."
@@ -247,9 +222,6 @@ def local_host_button_click(app_name: str, code: str) -> str:
 # Function to handle the "Ship" button click
 def ship_button_click(app_name: str, code: str) -> str:
     """Handles the "Ship" button click."""
-    # Get the app name from the user
-    app_name = gr.Textbox.get().strip()
-
     # Validate the app name
     if not app_name:
         return "Please enter a valid app name."
@@ -262,9 +234,17 @@ def ship_button_click(app_name: str, code: str) -> str:
 
 # --- Gradio Interface ---
 
-with gr.Blocks(theme='ParityError/Interstellar') as demo:
+with gr.Blocks(css=customCSS, theme='ParityError/Interstellar') as demo:
+    gr.Markdown(
+        """
+        # AI-Powered Code Generation and Web App Creation
+        This application allows you to interact with an AI agent cluster to generate code and create web apps.
+        """
+    )
+
     # --- Agent Selection ---
     with gr.Row():
+        gr.Markdown("## Select Your Agent Cluster")
         for agent_name, agent_data in agent_roles.items():
             button = gr.Button(agent_name, variant="secondary")
             textbox = gr.Textbox(agent_data["description"], interactive=False)
@@ -272,47 +252,48 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
 
     # --- Chat Interface ---
     with gr.Row():
+        gr.Markdown("## Chat with the AI")
         chatbot = gr.Chatbot()
         chat_interface_input = gr.Textbox(label="Enter your message", placeholder="Ask me anything!")
-        chat_interface_output = gr.Textbox(label="Response", interactive=False)
 
         # Parameters
-        temperature_slider = gr.Slider(
-            label="Temperature",
-            value=DEFAULT_TEMPERATURE,
-            minimum=0.0,
-            maximum=1.0,
-            step=0.05,
-            interactive=True,
-            info="Higher values generate more diverse outputs",
-        )
-        max_new_tokens_slider = gr.Slider(
-            label="Maximum New Tokens",
-            value=DEFAULT_MAX_NEW_TOKENS,
-            minimum=64,
-            maximum=4096,
-            step=64,
-            interactive=True,
-            info="The maximum number of new tokens",
-        )
-        top_p_slider = gr.Slider(
-            label="Top-p (Nucleus Sampling)",
-            value=DEFAULT_TOP_P,
-            minimum=0.0,
-            maximum=1,
-            step=0.05,
-            interactive=True,
-            info="Higher values sample more low-probability tokens",
-        )
-        repetition_penalty_slider = gr.Slider(
-            label="Repetition Penalty",
-            value=DEFAULT_REPETITION_PENALTY,
-            minimum=1.0,
-            maximum=2.0,
-            step=0.05,
-            interactive=True,
-            info="Penalize repeated tokens",
-        )
+        with gr.Accordion("Advanced Parameters", open=False):
+            temperature_slider = gr.Slider(
+                label="Temperature",
+                value=DEFAULT_TEMPERATURE,
+                minimum=0.0,
+                maximum=1.0,
+                step=0.05,
+                interactive=True,
+                info="Higher values generate more diverse outputs",
+            )
+            max_new_tokens_slider = gr.Slider(
+                label="Maximum New Tokens",
+                value=DEFAULT_MAX_NEW_TOKENS,
+                minimum=64,
+                maximum=4096,
+                step=64,
+                interactive=True,
+                info="The maximum number of new tokens",
+            )
+            top_p_slider = gr.Slider(
+                label="Top-p (Nucleus Sampling)",
+                value=DEFAULT_TOP_P,
+                minimum=0.0,
+                maximum=1,
+                step=0.05,
+                interactive=True,
+                info="Higher values sample more low-probability tokens",
+            )
+            repetition_penalty_slider = gr.Slider(
+                label="Repetition Penalty",
+                value=DEFAULT_REPETITION_PENALTY,
+                minimum=1.0,
+                maximum=2.0,
+                step=0.05,
+                interactive=True,
+                info="Penalize repeated tokens",
+            )
 
         # Submit Button
         submit_button = gr.Button("Submit")
@@ -323,7 +304,6 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             inputs=[
                 chat_interface_input,
                 chatbot,
-                get_agent_cluster,
                 temperature_slider,
                 max_new_tokens_slider,
                 top_p_slider,
@@ -331,12 +311,12 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             ],
             outputs=[
                 chatbot,
-                chat_interface_output,
             ],
         )
 
     # --- Web App Creation ---
     with gr.Row():
+        gr.Markdown("## Create Your Web App")
         app_name_input = gr.Textbox(label="App Name", placeholder="Enter your app name")
         code_output = gr.Textbox(label="Code", interactive=False)
         create_web_app_button = gr.Button("Create Web App")
@@ -347,7 +327,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         # Web App Creation Logic
         create_web_app_button.click(
             create_web_app_button_click,
-            inputs=[code_output],
+            inputs=[app_name_input, code_output],
             outputs=[gr.Textbox(label="Status", interactive=False)],
         )
 
@@ -373,9 +353,9 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         )
 
     # --- Connect Chat Output to Code Output ---
-    chat_interface_output.change(
-        lambda x: x,
-        inputs=[chat_interface_output],
+    chatbot.change(
+        lambda x: x[-1][1] if x else "",
+        inputs=[chatbot],
         outputs=[code_output],
     )