Update main.py
Browse filesRemoved risky function which re-creates project folder structure
main.py
CHANGED
|
@@ -39,30 +39,6 @@ def run_flask_frontend():
|
|
| 39 |
subprocess.run(["flask", "run", "--host=0.0.0.0", "--port=7860"],
|
| 40 |
env={**os.environ, "FLASK_APP": "buffalo_rag/frontend/flask_app.py"})
|
| 41 |
|
| 42 |
-
def create_project_structure():
|
| 43 |
-
"""Create the project folder structure."""
|
| 44 |
-
# Main directories
|
| 45 |
-
directories = [
|
| 46 |
-
"buffalo_rag",
|
| 47 |
-
"buffalo_rag/scraper",
|
| 48 |
-
"buffalo_rag/embeddings",
|
| 49 |
-
"buffalo_rag/vector_store",
|
| 50 |
-
"buffalo_rag/model",
|
| 51 |
-
"buffalo_rag/api",
|
| 52 |
-
"buffalo_rag/frontend",
|
| 53 |
-
"buffalo_rag/utils",
|
| 54 |
-
"data",
|
| 55 |
-
"data/raw",
|
| 56 |
-
"data/processed",
|
| 57 |
-
"data/embeddings",
|
| 58 |
-
]
|
| 59 |
-
|
| 60 |
-
# Create directories
|
| 61 |
-
for directory in directories:
|
| 62 |
-
os.makedirs(directory, exist_ok=True)
|
| 63 |
-
|
| 64 |
-
print("Project structure created!")
|
| 65 |
-
|
| 66 |
def build_react_frontend():
|
| 67 |
"""Build the React frontend."""
|
| 68 |
print("Building React frontend...")
|
|
@@ -118,7 +94,6 @@ def wait_for_server(url, timeout=30, interval=1):
|
|
| 118 |
|
| 119 |
def main():
|
| 120 |
parser = argparse.ArgumentParser(description="BuffaloRAG - AI Assistant for UB International Students")
|
| 121 |
-
parser.add_argument("--setup", action="store_true", help="Create project structure")
|
| 122 |
parser.add_argument("--flask-setup", action="store_true", help="Setup Flask frontend")
|
| 123 |
parser.add_argument("--scrape", action="store_true", help="Run web scraper")
|
| 124 |
parser.add_argument("--build", action="store_true", help="Build embeddings")
|
|
@@ -128,10 +103,7 @@ def main():
|
|
| 128 |
parser.add_argument("--run", action="store_true", help="Run frontend & backend servers")
|
| 129 |
|
| 130 |
args = parser.parse_args()
|
| 131 |
-
|
| 132 |
-
if args.setup or args.all:
|
| 133 |
-
create_project_structure()
|
| 134 |
-
|
| 135 |
if args.flask_setup or args.all or args.run:
|
| 136 |
# Run the Flask setup script
|
| 137 |
from setup_flask_templates import setup_flask_templates
|
|
|
|
| 39 |
subprocess.run(["flask", "run", "--host=0.0.0.0", "--port=7860"],
|
| 40 |
env={**os.environ, "FLASK_APP": "buffalo_rag/frontend/flask_app.py"})
|
| 41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
def build_react_frontend():
|
| 43 |
"""Build the React frontend."""
|
| 44 |
print("Building React frontend...")
|
|
|
|
| 94 |
|
| 95 |
def main():
|
| 96 |
parser = argparse.ArgumentParser(description="BuffaloRAG - AI Assistant for UB International Students")
|
|
|
|
| 97 |
parser.add_argument("--flask-setup", action="store_true", help="Setup Flask frontend")
|
| 98 |
parser.add_argument("--scrape", action="store_true", help="Run web scraper")
|
| 99 |
parser.add_argument("--build", action="store_true", help="Build embeddings")
|
|
|
|
| 103 |
parser.add_argument("--run", action="store_true", help="Run frontend & backend servers")
|
| 104 |
|
| 105 |
args = parser.parse_args()
|
| 106 |
+
|
|
|
|
|
|
|
|
|
|
| 107 |
if args.flask_setup or args.all or args.run:
|
| 108 |
# Run the Flask setup script
|
| 109 |
from setup_flask_templates import setup_flask_templates
|