Update app.py

app.py

@@ -1,13 +1,20 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
 import google.generativeai as genai
 import subprocess
 import re
 
 # 🔑 Configure Gemini API
-genai.configure(api_key="AIzaSyAdvERSmk7q02T2pLNPoERqDD-QPuW4RTs")
+genai.configure(api_key="YOUR_API_KEY")
 
 # Pick a model
 model = genai.GenerativeModel("gemini-1.5-flash")
 
+app = FastAPI()
+
+class TaskRequest(BaseModel):
+    task: str
+
 def extract_code(response_text):
     """Extracts Python code from Gemini's response (removes ``` fences)."""
     code_match = re.findall(r"```(?:python)?\n(.*?)```", response_text, re.DOTALL)
@@ -30,32 +37,21 @@ def run_code(code):
 
 def recursive_ai_executor(task, depth=1):
     """Recursive AI Executor with auto-run + visible output."""
-    print(f"\n🔹 Task (Level {depth}): {task}")
-    
-    # Ask Gemini to generate code + usage
     prompt = f"Write Python code to {task}. Always include a sample call (like print(...)) so output is shown."
     response = model.generate_content(prompt)
     code = extract_code(response.text)
-    
-    print("\n📜 Generated Code:\n", code)
     output = run_code(code)
-    print("\n📊 Execution Output:\n", output)
-    
-    # If no useful output, refine recursively
-    if "Traceback" in output or output.strip() == "":
-        if depth < 3:
-            print("\n⚡ Refining task...")
-            return recursive_ai_executor(f"Fix the error: {output}", depth+1)
-        else:
-            print("\n❌ Stopping after 3 attempts.")
-    return output
-
-
-# 🚀 Interactive loop
-print("🤖 Recursive AI Executor (type 'exit' to quit)\n")
-while True:
-    user_task = input("Enter your Python task: ")
-    if user_task.lower() in ["exit", "quit"]:
-        print("👋 Exiting AI Executor.")
-        break
-    recursive_ai_executor(user_task)
+
+    # Retry if failure
+    if ("Traceback" in output or output.strip() == "") and depth < 3:
+        return recursive_ai_executor(f"Fix the error: {output}", depth+1)
+
+    return {"generated_code": code, "execution_output": output}
+
+@app.post("/run-task")
+def run_task(request: TaskRequest):
+    return recursive_ai_executor(request.task)
+
+@app.get("/")
+def home():
+    return {"message": "🚀 Recursive AI Executor is running on Hugging Face!"}