Update app.py

app.py

@@ -1,13 +1,11 @@
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from langchain_community.llms import HuggingFacePipeline
 from langchain.agents import initialize_agent, Tool
-from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
-from langchain.llms import HuggingFacePipeline
 import json
 import subprocess
-import os
 import logging
 
 # Configure logging
@@ -24,7 +22,6 @@ hf_pipeline = pipeline(
     "text-generation",
     model=model,
     tokenizer=tokenizer,
-    device=0 if torch.cuda.is_available() else -1,
     max_length=500,
     temperature=0.7,
 )
@@ -110,64 +107,44 @@ def multi_agent_workflow(requirements: str) -> str:
     Returns:
         str: Generated app code and API details.
     """
-    global api_details
-
-    # Step 1: Task Decomposition
     try:
+        # Step 1: Task Decomposition
         task_decomposition = ui_designer_agent.run(
             f"Break down the following app requirements into smaller tasks: {requirements}"
         )
         tasks = json.loads(task_decomposition)["tasks"]
-    except Exception as e:
-        logger.error(f"Task decomposition failed: {str(e)}")
-        return f"Task decomposition failed: {str(e)}"
+        logger.info(f"Tasks: {tasks}")
 
-    # Step 2: Code Generation
-    try:
+        # Step 2: Code Generation
         ui_code = ui_designer_agent.run(f"Generate the UI code for: {tasks[0]}")
         backend_code = backend_developer_agent.run(f"Generate the backend code for: {tasks[1]}")
-    except Exception as e:
-        logger.error(f"Code generation failed: {str(e)}")
-        return f"Code generation failed: {str(e)}"
+        logger.info(f"UI Code: {ui_code}")
+        logger.info(f"Backend Code: {backend_code}")
 
-    # Step 3: Code Formatting
-    try:
+        # Step 3: Code Formatting
         formatted_ui_code = ui_designer_agent.run(f"Format the following code: {ui_code}")
         formatted_backend_code = backend_developer_agent.run(f"Format the following code: {backend_code}")
-    except Exception as e:
-        logger.error(f"Code formatting failed: {str(e)}")
-        return f"Code formatting failed: {str(e)}"
+        logger.info(f"Formatted UI Code: {formatted_ui_code}")
+        logger.info(f"Formatted Backend Code: {formatted_backend_code}")
 
-    # Step 4: Integration
-    try:
+        # Step 4: Integration
         combined_code = f"{formatted_ui_code}\n\n{formatted_backend_code}"
-    except Exception as e:
-        logger.error(f"Code integration failed: {str(e)}")
-        return f"Code integration failed: {str(e)}"
+        logger.info(f"Combined Code: {combined_code}")
 
-    # Step 5: Testing
-    try:
+        # Step 5: Testing
         test_results = qa_engineer_agent.run(f"Test the following app: {combined_code}")
-    except Exception as e:
-        logger.error(f"Testing failed: {str(e)}")
-        return f"Testing failed: {str(e)}"
+        logger.info(f"Test Results: {test_results}")
 
-    # Step 6: Deployment
-    try:
+        # Step 6: Deployment
         deployment_status = devops_engineer_agent.run(f"Deploy the following app: {combined_code}")
-    except Exception as e:
-        logger.error(f"Deployment failed: {str(e)}")
-        return f"Deployment failed: {str(e)}"
+        logger.info(f"Deployment Status: {deployment_status}")
 
-    # Step 7: API Documentation
-    try:
+        # Step 7: API Documentation
         api_details = backend_developer_agent.run(f"Generate API details for: {combined_code}")
-    except Exception as e:
-        logger.error(f"API documentation failed: {str(e)}")
-        return f"API documentation failed: {str(e)}"
+        logger.info(f"API Details: {api_details}")
 
-    # Return the results
-    return f"""
+        # Return the results
+        return f"""
 Generated App Code:
 {combined_code}
 
@@ -180,6 +157,9 @@ Deployment Status:
 API Details:
 {api_details}
 """
+    except Exception as e:
+        logger.error(f"Workflow failed: {str(e)}")
+        return f"Workflow failed: {str(e)}"
 
 # Gradio Interface
 def app_generator(requirements: str):