Create app.py

app.py

@@ -0,0 +1,52 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import openai  # Can be replaced with another AI API
+import os
+import subprocess
+
+app = FastAPI()
+
+# AI Model Selection
+MODEL_MAP = {
+    "gpt": "gpt-4",
+    "gemini": "gemini-1",
+    "custom": "http://custom-llm-api"
+}
+
+class TestRequest(BaseModel):
+    model: str
+    prompt: str
+
+@app.post("/generate_test")
+def generate_test(request: TestRequest):
+    """Generate test script using the selected AI model."""
+    model_choice = request.model
+    prompt = request.prompt
+    
+    model_api = MODEL_MAP.get(model_choice, "gpt-4")
+    try:
+        response = openai.ChatCompletion.create(
+            model=model_api,
+            messages=[{"role": "system", "content": prompt}]
+        )
+        test_script = response["choices"][0]["message"]["content"]
+        
+        # Save the generated test script
+        with open("generated_test.py", "w") as f:
+            f.write(test_script)
+
+        return {"message": "Test generated", "script": test_script}
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/run_tests")
+def run_tests():
+    """Execute AI-generated test scripts and return results."""
+    try:
+        result = subprocess.run(["pytest", "generated_test.py", "--json-report"], capture_output=True, text=True)
+        return {"output": result.stdout, "error": result.stderr}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# Run with: uvicorn app:app --host 0.0.0.0 --port 7860