Update app.py

app.py

@@ -8,19 +8,22 @@ import time
 import shutil
 from typing import Dict, Tuple
 
-# Constants
+# --- Constants ---
+
 API_URL = "https://api-inference.huggingface.co/models/"
 MODEL_NAME = "mistralai/Mixtral-8x7B-Instruct-v0.1"  # Replace with your desired model
+
+# Chat Interface Parameters
 DEFAULT_TEMPERATURE = 0.9
 DEFAULT_MAX_NEW_TOKENS = 2048
 DEFAULT_TOP_P = 0.95
 DEFAULT_REPETITION_PENALTY = 1.2
+
+# Local Server
 LOCAL_HOST_PORT = 7860
 
-# Initialize the InferenceClient
-client = InferenceClient(MODEL_NAME)
+# --- Agent Roles ---
 
-# Define agent roles and their initial states
 agent_roles: Dict[str, Dict[str, bool]] = {
     "Web Developer": {"description": "A master of front-end and back-end web development.", "active": False},
     "Prompt Engineer": {"description": "An expert in crafting effective prompts for AI models.", "active": False},
@@ -29,10 +32,9 @@ agent_roles: Dict[str, Dict[str, bool]] = {
     "AI-Powered Code Assistant": {"description": "An AI assistant that can help with coding tasks and provide code snippets.", "active": False},
 }
 
-# Initialize the selected agent
-selected_agent = list(agent_roles.keys())[0]
+# --- Initial Prompt ---
 
-# Initial prompt for the selected agent
+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.
@@ -56,14 +58,17 @@ Please make sure to answer in the language used by the user. If the user asks in
 But, you can go ahead and search in English, especially for programming-related questions. PLEASE MAKE SURE TO ALWAYS SEARCH IN ENGLISH FOR THOSE.
 """
 
-# Custom CSS for the chat interface
+# --- Custom CSS ---
+
 customCSS = """
-#component-7 { # dies ist die Standardelement-ID des Chatkomponenten
-  height: 1600px; # passen Sie die Höhe nach Bedarf an
+#component-7 { 
+  height: 1600px; 
   flex-grow: 4;
 }
 """
 
+# --- Functions ---
+
 # Function to toggle the active state of an agent
 def toggle_agent(agent_name: str) -> str:
     """Toggles the active state of an agent."""
@@ -151,7 +156,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 = DEFAULT_TEMPERATURE, max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS, top_p: float = DEFAULT_TOP_P, repetition_penalty: float = DEFAULT_REPETITION_PENALTY) -> Tuple[str, str]:
+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]:
     """Handles user input and generates responses."""
     if message.startswith("python"): 
         # User entered code, execute it 
@@ -246,22 +251,23 @@ def ship_button_click(app_name: str, code: str) -> str:
     # Return a success message
     return f"Web app '{app_name}' shipped successfully!"
 
-# Create the Gradio interface
+# --- Gradio Interface ---
+
 with gr.Blocks(theme='ParityError/Interstellar') as demo:
-    # Agent selection area
+    # --- Agent Selection ---
     with gr.Row():
         for agent_name, agent_data in agent_roles.items():
             button = gr.Button(agent_name, variant="secondary")
             textbox = gr.Textbox(agent_data["description"], interactive=False)
             button.click(toggle_agent, inputs=[button], outputs=[textbox])
 
-    # Chat interface area
+    # --- Chat Interface ---
     with gr.Row():
         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 for the chat interface
+        # Parameters
         temperature_slider = gr.Slider(
             label="Temperature",
             value=DEFAULT_TEMPERATURE,
@@ -299,15 +305,15 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             info="Penalize repeated tokens",
         )
 
-        # Submit button for the chat interface
+        # Submit Button
         submit_button = gr.Button("Submit")
 
-        # Create the chat interface
+        # Chat Interface Logic
         submit_button.click(
             chat_interface,
             inputs=[
-                chat_interface_input,
-                chatbot,
+                chat_interface_input._id,
+                chatbot._id,
                 get_agent_cluster,
                 temperature_slider,
                 max_new_tokens_slider,
@@ -320,7 +326,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             ],
         )
 
-    # Web app creation area
+    # --- Web App Creation ---
     with gr.Row():
         app_name_input = gr.Textbox(label="App Name", placeholder="Enter your app name")
         code_output = gr.Textbox(label="Code", interactive=False)
@@ -329,7 +335,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
         local_host_button = gr.Button("Local Host")
         ship_button = gr.Button("Ship")
 
-        # Create the web app creation interface
+        # Web App Creation Logic
         create_web_app_button.click(
             create_web_app_button_click,
             inputs=[code_output],
@@ -357,12 +363,12 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
             outputs=[gr.Textbox(label="Status", interactive=False)],
         )
 
-    # Connect the chat interface output to the code output
+    # --- Connect Chat Output to Code Output ---
     chat_interface_output.change(
         lambda x: x,
         inputs=[chat_interface_output],
         outputs=[code_output],
     )
 
-    # Launch the Gradio interface
+    # --- Launch Gradio ---
     demo.queue().launch(debug=True)