upload functoins added to app.py

app.py

@@ -1,7 +1,61 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.middleware.cors import CORSMiddleware # Important for frontend
+import uuid
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
+# --- IMPORTANT: CORS Middleware ---
+# This allows your Netlify frontend to talk to this backend.
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Or specify your Netlify URL for production
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# In-memory session store
+SESSION_DATA = {}
+logger.info("Session store initialized.")
+
 @app.get("/")
-def greet_json():
-    return {"Welcome to OpenQuery"}
+def read_root():
+    return {"message": "Universal Data AI is running!"}
+
+@app.post("/upload")
+async def upload_file(file: UploadFile = File(...)):
+    """
+    Accepts a file, creates a session, and stores the file content in memory.
+    """
+    session_id = str(uuid.uuid4())
+    logger.info(f"New upload received. Creating session_id: {session_id}")
+
+    try:
+        # For now, we'll just store the content and filename.
+        # Later, this is where you'll call your parsing functions.
+        content = await file.read()
+        SESSION_DATA[session_id] = {
+            "filename": file.filename,
+            "content_type": file.content_type,
+            "data": content  # In a real app, you'd parse this
+        }
+        logger.info(f"File '{file.filename}' stored for session {session_id}")
+        return {"session_id": session_id, "filename": file.filename}
+    except Exception as e:
+        logger.error(f"Error processing file for session {session_id}: {e}")
+        raise HTTPException(status_code=500, detail="Error processing file.")
+
+@app.get("/session/{session_id}")
+def get_session_data(session_id: str):
+    """
+    Retrieves the data stored for a given session.
+    """
+    if session_id not in SESSION_DATA:
+        raise HTTPException(status_code=404, detail="Session not found.")
+    # Return metadata, not the raw data, for privacy
+    return {"filename": SESSION_DATA[session_id].get("filename")}