|
|
""" |
|
|
Quick start script to run the RAG application. |
|
|
""" |
|
|
import subprocess |
|
|
import sys |
|
|
import os |
|
|
|
|
|
|
|
|
def check_dependencies(): |
|
|
"""Check if required dependencies are installed.""" |
|
|
try: |
|
|
import streamlit |
|
|
import fastapi |
|
|
import groq |
|
|
return True |
|
|
except ImportError: |
|
|
return False |
|
|
|
|
|
|
|
|
def install_dependencies(): |
|
|
"""Install dependencies from requirements.txt.""" |
|
|
print("Installing dependencies...") |
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"]) |
|
|
print("Dependencies installed successfully!") |
|
|
|
|
|
|
|
|
def check_env_file(): |
|
|
"""Check if .env file exists.""" |
|
|
if not os.path.exists(".env"): |
|
|
print("\n⚠️ Warning: .env file not found!") |
|
|
print("Creating .env from .env.example...") |
|
|
if os.path.exists(".env.example"): |
|
|
with open(".env.example", "r") as src: |
|
|
with open(".env", "w") as dst: |
|
|
dst.write(src.read()) |
|
|
print("✅ .env file created. Please edit it and add your Groq API key.") |
|
|
else: |
|
|
print("❌ .env.example not found. Please create .env manually.") |
|
|
return False |
|
|
return True |
|
|
|
|
|
|
|
|
def run_streamlit(): |
|
|
"""Run the Streamlit application.""" |
|
|
print("\n🚀 Starting Streamlit application...") |
|
|
print("📱 Open your browser to: http://localhost:8501") |
|
|
subprocess.run([sys.executable, "-m", "streamlit", "run", "streamlit_app.py"]) |
|
|
|
|
|
|
|
|
def run_api(): |
|
|
"""Run the FastAPI application.""" |
|
|
print("\n🚀 Starting FastAPI server...") |
|
|
print("📱 API available at: http://localhost:8000") |
|
|
print("📚 API docs at: http://localhost:8000/docs") |
|
|
subprocess.run([sys.executable, "api.py"]) |
|
|
|
|
|
|
|
|
def main(): |
|
|
"""Main function.""" |
|
|
print("=" * 50) |
|
|
print("RAG Capstone Project - Quick Start") |
|
|
print("=" * 50) |
|
|
|
|
|
|
|
|
if not check_dependencies(): |
|
|
print("\n📦 Installing dependencies...") |
|
|
install_dependencies() |
|
|
|
|
|
|
|
|
env_exists = check_env_file() |
|
|
|
|
|
if not env_exists: |
|
|
print("\n❌ Please configure your .env file before running the application.") |
|
|
return |
|
|
|
|
|
|
|
|
print("\nWhat would you like to run?") |
|
|
print("1. Streamlit Chat Interface (recommended)") |
|
|
print("2. FastAPI Backend") |
|
|
print("3. Both (requires two terminals)") |
|
|
|
|
|
choice = input("\nEnter your choice (1-3): ").strip() |
|
|
|
|
|
if choice == "1": |
|
|
run_streamlit() |
|
|
elif choice == "2": |
|
|
run_api() |
|
|
elif choice == "3": |
|
|
print("\n📌 To run both:") |
|
|
print("Terminal 1: python api.py") |
|
|
print("Terminal 2: streamlit run streamlit_app.py") |
|
|
print("\nStarting Streamlit in this terminal...") |
|
|
run_streamlit() |
|
|
else: |
|
|
print("Invalid choice. Exiting.") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
main() |
|
|
|
