oopere commited on
Commit
8103ea6
·
1 Parent(s): d5455f4

Formatted with isort.

Browse files

refactor: reorganize imports across multiple files for consistency

app.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
  import threading
3
  import time
 
4
  import uvicorn
5
  from optipfair_backend import app as fastapi_app
6
  from optipfair_frontend import create_interface
 
1
  import os
2
  import threading
3
  import time
4
+
5
  import uvicorn
6
  from optipfair_backend import app as fastapi_app
7
  from optipfair_frontend import create_interface
optipfair_backend.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastapi import FastAPI
2
  from fastapi.middleware.cors import CORSMiddleware # ← NUEVO: Para CORS
 
3
  from routers.visualize import router as visualize_router
4
 
5
  # Create FastAPI app with HF Spaces configuration
 
1
  from fastapi import FastAPI
2
  from fastapi.middleware.cors import CORSMiddleware # ← NUEVO: Para CORS
3
+
4
  from routers.visualize import router as visualize_router
5
 
6
  # Create FastAPI app with HF Spaces configuration
optipfair_frontend.py CHANGED
@@ -1,9 +1,10 @@
1
- import gradio as gr
2
- import requests
3
  import logging
4
  import os
5
  import tempfile
6
 
 
 
 
7
  # Configure logging
8
  logging.basicConfig(level=logging.INFO)
9
  logger = logging.getLogger(__name__)
 
 
 
1
  import logging
2
  import os
3
  import tempfile
4
 
5
+ import gradio as gr
6
+ import requests
7
+
8
  # Configure logging
9
  logging.basicConfig(level=logging.INFO)
10
  logger = logging.getLogger(__name__)
routers/visualize.py CHANGED
@@ -1,18 +1,14 @@
1
  # routers/visualize.py
2
- import os
3
  import logging
 
 
4
  from fastapi import APIRouter, HTTPException
5
  from fastapi.responses import FileResponse
6
- from schemas.visualize import (
7
- VisualizePCARequest,
8
- VisualizeMeanDiffRequest,
9
- VisualizeHeatmapRequest,
10
- )
11
- from utils.visualize_pca import (
12
- run_visualize_pca,
13
- run_visualize_mean_diff,
14
- run_visualize_heatmap,
15
- )
16
 
17
  logger = logging.getLogger(__name__)
18
  logger.setLevel(logging.INFO)
 
1
  # routers/visualize.py
 
2
  import logging
3
+ import os
4
+
5
  from fastapi import APIRouter, HTTPException
6
  from fastapi.responses import FileResponse
7
+
8
+ from schemas.visualize import (VisualizeHeatmapRequest,
9
+ VisualizeMeanDiffRequest, VisualizePCARequest)
10
+ from utils.visualize_pca import (run_visualize_heatmap,
11
+ run_visualize_mean_diff, run_visualize_pca)
 
 
 
 
 
12
 
13
  logger = logging.getLogger(__name__)
14
  logger.setLevel(logging.INFO)
schemas/visualize.py CHANGED
@@ -1,6 +1,7 @@
1
  # schemas/visualize.py
 
 
2
  from pydantic import BaseModel, field_validator
3
- from typing import List, Optional, Union, Tuple
4
 
5
 
6
  class VisualizePCARequest(BaseModel):
 
1
  # schemas/visualize.py
2
+ from typing import List, Optional, Tuple, Union
3
+
4
  from pydantic import BaseModel, field_validator
 
5
 
6
 
7
  class VisualizePCARequest(BaseModel):
utils/visualize_pca.py CHANGED
@@ -1,16 +1,16 @@
1
  # utils/visualize_pca.py
 
2
  import os
3
  import tempfile
4
- import logging
5
  from functools import lru_cache
6
- from typing import Tuple, Optional, Union, List
7
 
 
8
  import torch
9
- from optipfair.bias import visualize_pca, visualize_mean_differences, visualize_heatmap
 
10
  from transformers import AutoModelForCausalLM, AutoTokenizer
11
 
12
- import matplotlib
13
-
14
  matplotlib.use("Agg") # Use 'Agg' backend for non-GUI environments
15
 
16
  logger = logging.getLogger(__name__)
 
1
  # utils/visualize_pca.py
2
+ import logging
3
  import os
4
  import tempfile
 
5
  from functools import lru_cache
6
+ from typing import List, Optional, Tuple, Union
7
 
8
+ import matplotlib
9
  import torch
10
+ from optipfair.bias import (visualize_heatmap, visualize_mean_differences,
11
+ visualize_pca)
12
  from transformers import AutoModelForCausalLM, AutoTokenizer
13
 
 
 
14
  matplotlib.use("Agg") # Use 'Agg' backend for non-GUI environments
15
 
16
  logger = logging.getLogger(__name__)