Fixed app v2

app.py

@@ -1,4 +1,5 @@
-from fastapi import FastAPI, HTTPException, UploadFile, File
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 from llama_cpp import Llama
 
 # Initialize FastAPI app
@@ -13,20 +14,24 @@ try:
 except Exception as e:
     raise RuntimeError(f"Failed to load model: {e}")
 
-# Define the route for security log analysis with file upload
-@app.post("/analyze_security_logs")
-async def analyze_security_logs(file: UploadFile = File(...)):
-    try:
-        # Read the content of the uploaded file
-        log_data = await file.read()
-        log_data = log_data.decode("utf-8")
+# Define request model for log data
+class LogRequest(BaseModel):
+    log_data: str
+
+# Define response model
+class AnalysisResponse(BaseModel):
+    analysis: str
 
+# Define the route for security log analysis
+@app.post("/analyze_security_logs", response_model=AnalysisResponse)
+async def analyze_security_logs(request: LogRequest):
+    try:
         # Security-focused prompt
         prompt = (
             "Analyze the following network log data for any indicators of malicious activity, "
             "such as unusual IP addresses, unauthorized access attempts, data exfiltration, or anomalies. "
             "Provide details on potential threats, IPs involved, and suggest actions if any threats are detected.\n\n"
-            f"{log_data}"
+            f"{request.log_data}"
         )
         
         # Generate response from the model
@@ -41,7 +46,7 @@ async def analyze_security_logs(file: UploadFile = File(...)):
         
         # Extract and return the analysis text
         analysis_text = response["choices"][0]["message"]["content"]
-        return {"analysis": analysis_text}
+        return AnalysisResponse(analysis=analysis_text)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))