Update main.py

main.py

@@ -1,27 +1,24 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
+from app.model import create_model
+from app.database import connect_to_database
+from app.integration import integrate_data
 
 import os
 app = FastAPI()
 
 @app.get("/")
 async def root():
-    return {"message": "Hello World"}
-model = load_model()
-
-client = MongoClient('mongodb://localhost:27017/')
-db = client['']
-collection = db['text']
-
+    
+def main():
+    # Veritabanına bağlan
+    db_connection = connect_to_database()
 
+    # Modeli oluştur
+    model = create_model()
 
-class InputData(BaseModel):
-    title: str
-    keywords: str
-    subheadings: str
+    # Veriyi entegre et
+    integrate_data(db_connection, model)
 
-@app.post("/generate/")
-async def generate(input_data: InputData):
-    
-     result = model.predict(input_data)
-    return {"result": "generated_output"}
+if __name__ == "__main__":
+    main()