Update app.py

app.py

@@ -1,9 +1,7 @@
 import streamlit as st
 from huggingface_hub import InferenceClient
 import os
-import sys
 import pickle
-import json
 
 st.title("CODEFUSSION ☄")
 
@@ -72,6 +70,15 @@ class DataRetrievalTool(Tool):
         source = arguments.get("source", "https://example.com/data")
         return {"output": f"Data from {source}"}
 
+class TextGenerationTool(Tool):
+    def __init__(self):
+        super().__init__("Text Generation", "Generates human-like text based on a given prompt.")
+
+    def run(self, arguments):
+        # This is a simplified example, real implementation would use a text generation model
+        prompt = arguments.get("prompt", "Write a short story about a cat.")
+        return {"output": f"Generated text: {prompt}"}
+
 class CodeExecutionTool(Tool):
     def __init__(self):
         super().__init__("Code Execution", "Runs code snippets in various languages.")
@@ -155,7 +162,7 @@ class ImageAnalysisTool(Tool):
 
 # --- Agent Pool ---
 agent_pool = {
-    "IdeaIntake": Agent("IdeaIntake", "Idea Intake", [DataRetrievalTool(), CodeGenerationTool(), SentimentAnalysisTool(), TextGenerationTool()], knowledge_base=""),
+    "IdeaIntake": Agent("IdeaIntake", "Idea Intake", [DataRetrievalTool(), CodeGenerationTool(), TextGenerationTool()], knowledge_base=""),
     "CodeBuilder": Agent("CodeBuilder", "Code Builder", [CodeGenerationTool(), CodeDebuggingTool(), CodeOptimizationTool()], knowledge_base=""),
     "ImageCreator": Agent("ImageCreator", "Image Creator", [ImageGenerationTool(), ImageEditingTool()], knowledge_base=""),
 }
@@ -347,7 +354,7 @@ if st.session_state.chat_state == "normal":
         st.session_state.messages.append({"role": "user", "content": prompt})
         conversation_history = [(message["role"], message["content"]) for message in st.session_state.messages]
         
-        formated_text = format_promt(prompt, conversation_history, custom_instruction)
+        formated_text = format_prompt(prompt, conversation_history, custom_instruction)
         
         with st.chat_message("assistant"):
             # Select the appropriate model based on the user's choice
@@ -421,7 +428,7 @@ for agent_name, agent in agent_pool.items():
 
 st.subheader("Workflows")
 st.write("**App Build**")
-st.write(f"""  Description: {AppBuildWorkflow().description}")
+st.write(f"""  Description: {AppBuildWorkflow().description}""")
 st.write("**Website Build**")
 st.write(f"""  Description: {WebsiteBuildWorkflow().description}""")
 st.write("**Game Build**")
@@ -433,12 +440,12 @@ st.write(f"""  Description: {DevSandboxWorkflow().description}""")
 
 # --- Displaying Tool Definitions ---
 st.subheader("Tool Definitions")
-for tool_class in [CodeGenerationTool, DataRetrievalTool, CodeExecutionTool, CodeDebuggingTool, CodeSummarizationTool, CodeTranslationTool, CodeOptimizationTool, CodeDocumentationTool, ImageGenerationTool, ImageEditingTool, ImageAnalysisTool]:
+for tool_class in [CodeGenerationTool, DataRetrievalTool, CodeExecutionTool, CodeDebuggingTool, CodeSummarizationTool, CodeTranslationTool, CodeOptimizationTool, CodeDocumentationTool, ImageGenerationTool, ImageEditingTool, ImageAnalysisTool, TextGenerationTool]:
     tool = tool_class()
     st.write(f"**{tool.name}**")
     st.write(f"  Description: {tool.description}")
-
 # --- Displaying Example Output ---
+
 st.subheader("Example Output")
 code_generation_tool = CodeGenerationTool()
 st.write(f"""Code Generation Tool Output: {code_generation_tool.run({'language': 'python', 'code': 'print('Hello, World!')'})}""")