Files
AI-Stock-Trader/WebServer/Controllers/PythonInterop.cs
T
2026-03-08 19:28:10 -07:00

55 lines
1.9 KiB
C#

using Python.Runtime;
namespace Controllers.PythonInterop {
public class AIModule {
public AIModule(string PythonPathBase = "/usr/local/", string PythonVersion = "python3.11") {
// Use the user provided python runner
Runtime.PythonDLL = PythonPathBase + $"lib/lib{PythonVersion}.so";
// Use our local environment for the python libraries
PythonEngine.PythonHome = PythonPathBase;
// Include all the paths for python packages most importantly our venv
PythonEngine.PythonPath = $"{PythonPathBase}lib/{PythonVersion}:{PythonPathBase}lib/{PythonVersion}/lib-dynload:{PythonPathBase}lib/{PythonVersion}/site-packages:{Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "AIPython")}";
// Initiilize python
PythonEngine.Initialize();
// Needed because C# calls the python from each connections worker thread
PythonEngine.BeginAllowThreads();
}
public void PullAI() {
using (Py.GIL()) {
dynamic datapuller = Py.Import("datapuller");
using (datapuller.pull()){ }
}
}
public void TrainAI() {
using (Py.GIL()) {
dynamic trainer = Py.Import("ai-trainer");
using (trainer.TrainAI()){ }
}
}
// Return ( Error, Signal )
public (string, int) PredictAI(string StockSymbol) {
try {
using (Py.GIL()) {
dynamic predictor = Py.Import("ai-predictor");
using (dynamic x = predictor.Predict(StockSymbol)) {
int result = (int)x;
return ("", result);
}
}
} catch (Exception ex) {
return (ex.ToString(), 0);
}
}
}
}