A newer version of the Gradio SDK is available:
5.37.0
metadata
license: mit
sdk: gradio
license: mit
π§ Disease Symptom Prediction AI
AI-powered tool that predicts possible diseases based on symptoms using a trained machine learning model and provides chatbot-style suggestions.
π Project Structure
disease-symptom-prediction/
βββ data/ # Excel dataset (not uploaded)
β βββ README.md # Instructions for downloading from Kaggle
β
βββ src/ # Source code
β βββ data_preprocessing.py # Cleans & prepares dataset
β βββ model.py # Trains and saves the ML model
β βββ predict.py # Predicts disease from symptoms
β βββ chatbot.py # Symptom-checker chatbot logic
β
βββ app/ # App interface
β βββ gradio_app.py # Gradio-based web UI
β
βββ models/ # Trained model files
β βββ rf_model.pkl
β
βββ notebooks/ # Jupyter notebooks for analysis
β βββ 1_data_exploration.ipynb
β βββ 2_preprocessing_test.ipynb
β βββ 3_model_testing.ipynb
β
βββ requirements.txt # Python dependencies
βββ .gitignore
βββ README.md
βοΈ Installation
1. Clone the repository
git clone https://github.com/your-username/disease-symptom-prediction.git
cd disease-symptom-prediction
2. Create virtual environment (optional but recommended)
python -m venv venv
source venv/bin/activate # On Windows: venv\Scripts\activate
3. Install dependencies
pip install -r requirements.txt
π Run the App
python app/gradio_app.py
The app will start locally and you can open it in your browser at http://localhost:7860
.
π€ Features
- β Predict disease from symptoms
- π§ Trained on structured healthcare dataset
- π¬ Chatbot-style health advice
- π₯οΈ User-friendly Gradio interface
π¦ Dataset Note (Click to expand)
We use a healthcare dataset available on Kaggle. Due to licensing, it is not uploaded in this repo. Please download it manually and place it in the data/
directory.
πͺͺ License
This project is licensed under the MIT License - see the LICENSE file for details.
π Credits
Created by Mate β feel free to fork, star, or contribute.