arahrooh commited on
Commit
122aba7
·
1 Parent(s): 589ce95

Fix: Add import error handling to prevent module load failure

Browse files
Files changed (1) hide show
  1. app.py +14 -2
app.py CHANGED
@@ -28,8 +28,20 @@ import logging
28
  import textstat
29
  import torch
30
 
31
- # Import from bot.py
32
- from bot import RAGBot, parse_args, Chunk
 
 
 
 
 
 
 
 
 
 
 
 
33
 
34
  # Set up logging first (before any logger usage)
35
  logging.basicConfig(level=logging.INFO)
 
28
  import textstat
29
  import torch
30
 
31
+ # Import from bot.py - wrap in try/except to handle import errors gracefully
32
+ try:
33
+ from bot import RAGBot, parse_args, Chunk
34
+ BOT_AVAILABLE = True
35
+ except ImportError as e:
36
+ logger.error(f"Failed to import bot module: {e}")
37
+ BOT_AVAILABLE = False
38
+ # Create dummy classes so the module can still load
39
+ class RAGBot:
40
+ pass
41
+ class Chunk:
42
+ pass
43
+ def parse_args():
44
+ return None
45
 
46
  # Set up logging first (before any logger usage)
47
  logging.basicConfig(level=logging.INFO)