Update app.py

app.py

@@ -7,7 +7,6 @@ from abc import ABC, abstractmethod
 import gradio as gr
 from langchain_community.llms import HuggingFaceHub
 from dotenv import load_dotenv
-from langchain.llms import HuggingFaceHub
 from langchain.agents import initialize_agent, AgentType
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
@@ -50,21 +49,6 @@ def load_config() -> Dict:
     config['api_key'] = os.getenv('HUGGINGFACE_API_KEY', config['api_key'])
     return config
 
-@spaces.GPU()
-def stream_chat(
-    message: str, 
-    history: list,
-    tools: str,
-    temperature: float = 0.3, 
-    max_tokens: int = 1024, 
-):
-    # Define persona attributes
-    persona_greeting = "Hello! I'm your friendly assistant. How can I help you today?"
-    
-    # Start conversation with persona greeting
-    if not history:
-        history.append(("AI", persona_greeting))  # Assuming history is a list of tuples
-
 def setup_logging() -> logging.Logger:
     """Set up logging configuration."""
     logging.basicConfig(
@@ -184,11 +168,16 @@ class Agent:
             logger.error(f"Error initializing agent: {e}")
             raise AgentInitializationError(f"Failed to initialize agent: {e}")
 
-    async def act(self, prompt: str, context: str) -> str:
+    async def act(self, prompt: str, context: str, mode: str) -> str:
         """Perform an action based on the given prompt and context."""
         self.memory.append((prompt, context))
         try:
-            action = await self.agent.arun(prompt, context)
+            if mode == "full":
+                action = await self.agent.arun(prompt, context)
+            elif mode == "half":
+                action = f"Please follow these instructions: {prompt}. Then, let me know what you did."
+            else:  # mode == "none"
+                action = "I'm here if you need assistance. Just ask!"
             return action
         except Exception as e:
             logger.error(f"Error during agent action: {e}")
@@ -198,7 +187,7 @@ class Agent:
         return f"Agent: {self.name} (Role: {self.role})"
 
 # Main application functions
-async def run(message: str, history: List[Tuple[str, str]]) -> str:
+async def run(message: str, history: List[Tuple[str, str]], mode: str) -> str:
     """Process user input and generate a response using the agent system."""
     agent = Agent(
         name="CodeFusion",
@@ -207,7 +196,7 @@ async def run(message: str, history: List[Tuple[str, str]]) -> str:
     )
     context = "\n".join([f"Human: {h[0]}\nAI: {h[1]}" for h in history])
     try:
-        response = await agent.act(message, context)
+        response = await agent.act(message, context, mode)
         return response
     except Exception as e:
         logger.error(f"Error processing request: {e}")
@@ -224,46 +213,14 @@ async def main():
     gr.ChatInterface(
         fn=run,
         title="CodeFusion: Your AI Coding Assistant",
-        description="Ask me about code generation, explanation, debugging, or any other coding task!",
+        description="Ask me about code generation, explanation, debugging, or any other coding task! Choose a mode: full autonomy, 50% engagement, or user-referred assistance.",
         examples=examples,
         theme="default"
     ).launch()
 
-    async def test_code_generation():
-        tool = CodeGenerationTool()
-        result = await tool.run({"language": "python", "code_description": "function to add two numbers"})
-        assert "def" in result["output"], "Code generation failed to produce a function"
-        print("Code Generation Test: Passed")
-
-    async def test_code_explanation():
-        tool = CodeExplanationTool()
-        result = await tool.run({"code": "def factorial(n):\n    return 1 if n == 0 else n * factorial(n-1)"})
-        assert "recursive" in result["output"].lower(), "Code explanation failed to mention recursion"
-        print("Code Explanation Test: Passed")
-
-    async def test_debugging():
-        tool = DebuggingTool()
-        result = await tool.run({"code": "def divide(a, b):\n    return a / b", "error_message": "ZeroDivisionError"})
-        assert "zero" in result["output"].lower(), "Debugging failed to address division by zero"
-        print("Debugging Test: Passed")
-
-    async def test_agent():
-        agent = Agent("TestAgent", "Tester", [CodeGenerationTool(), CodeExplanationTool(), DebuggingTool()])
-        result = await agent.act("Generate a Python function to calculate the square of a number", "")
-        assert "def" in result and "return" in result, "Agent failed to generate a proper function"
-        print("Agent Test: Passed")
-
-    async def run_all_tests():
-        await test_code_generation()
-        await test_code_explanation()
-        await test_debugging()
-        await test_agent()
-
-    asyncio.run(run_all_tests())
-
 if __name__ == "__main__":
     import sys
     if len(sys.argv) > 1 and sys.argv[1] == "--test":
         run_tests()
     else:
-        asyncio.run(main())
+        asyncio.run(main())