Update app.py

app.py

@@ -422,10 +422,101 @@ examples = [
 "Seamless Web Assembly Embedding"]
 ]
 
-gr.ChatInterface(
-    fn=run,
-    chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
-    title="Mixtral 46.7B\nMicro-Agent\nInternet Search <br> development test",
-    examples=examples,
-    concurrency_limit=20,
-).launch(show_api=False)
+def parse_action(line):
+    action_name, action_input = line.strip("action: ").split("=")
+    action_input = action_input.strip()
+    return action_name, action_input
+
+def get_file_tree(path):
+    """
+    Recursively explores a directory and returns a nested dictionary representing its file tree.
+    """
+    tree = {}
+    for item in os.listdir(path):
+        item_path = os.path.join(path, item)
+        if os.path.isdir(item_path):
+            tree[item] = get_file_tree(item_path)
+        else:
+            tree[item] = None
+    return tree
+
+def display_file_tree(tree, indent=0):
+    """
+    Prints a formatted representation of the file tree.
+    """
+    for name, subtree in tree.items():
+        print(f"{'  ' * indent}{name}")
+        if subtree is not None:
+            display_file_tree(subtree, indent + 1)
+
+def project_explorer(path):
+    """
+    Displays the file tree of a given path in a Streamlit app.
+    """
+    tree = get_file_tree(path)
+    display_file_tree(tree)
+
+def chat_app_logic(message, history, purpose, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty):
+    """
+    Handles the chat app logic, including running the model and displaying the output.
+    """
+    if message:
+        # Run the model and get the response
+        response = generate(
+            prompt=message,
+            history=history,
+            agent_name=agent_name,
+            sys_prompt=sys_prompt,
+            temperature=temperature,
+            max_new_tokens=max_new_tokens,
+            top_p=top_p,
+            repetition_penalty=repetition_penalty,
+        )
+        history.append((message, response))
+        return history
+
+    return history
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("## FragMixt")
+        gr.Markdown("### Agents w/ Agents")
+
+        # Chat Interface
+        chatbot = gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel")
+        chatbot.load(examples)
+
+        # Input Components
+        message = gr.Textbox(label="Enter your message", placeholder="Ask me anything!")
+        purpose = gr.Textbox(label="Purpose", placeholder="What is the purpose of this interaction?")
+        agent_name = gr.Dropdown(label="Agents", choices=[s for s in agents], value=agents[0], interactive=True)
+        sys_prompt = gr.Textbox(label="System Prompt", max_lines=1, interactive=True)
+        temperature = gr.Slider(label="Temperature", value=0.9, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Higher values produce more diverse outputs")
+        max_new_tokens = gr.Slider(label="Max new tokens", value=1048*10, minimum=0, maximum=1048*10, step=64, interactive=True, info="The maximum numbers of new tokens")
+        top_p = gr.Slider(label="Top-p (nucleus sampling)", value=0.90, minimum=0.0, maximum=1, step=0.05, interactive=True, info="Higher values sample more low-probability tokens")
+        repetition_penalty = gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Penalize repeated tokens")
+
+        # Button to submit the message
+        submit_button = gr.Button(value="Send")
+
+        # Project Explorer Tab
+        with gr.Tab("Project Explorer"):
+            project_path = gr.Textbox(label="Project Path", placeholder="/path/to/project")
+            explore_button = gr.Button(value="Explore")
+            project_output = gr.Textbox(label="File Tree", lines=20)
+
+        # Chat App Logic Tab
+        with gr.Tab("Chat App"):
+            history = gr.State([])
+
+            # Connect components to the chat app logic
+            submit_button.click(chat_app_logic, inputs=[message, history, purpose, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty], outputs=chatbot)
+            message.submit(chat_app_logic, inputs=[message, history, purpose, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty], outputs=chatbot)
+
+        # Connect components to the project explorer
+        explore_button.click(project_explorer, inputs=project_path, outputs=project_output)
+
+    demo.launch(show_api=False)
+
+if __name__ == "__main__":
+    main()