msIntui commited on
Commit
690b5e4
·
1 Parent(s): 58bdeac

Fix merge conflict in config.py

Browse files
Files changed (2) hide show
  1. config.py +0 -10
  2. gradioChatApp.py +20 -28
config.py CHANGED
@@ -89,7 +89,6 @@ class JunctionConfig(BaseConfig):
89
  radius: int = 5
90
  angle_threshold_lb: float = 15.0
91
  angle_threshold_ub: float = 75.0
92
- <<<<<<< HEAD
93
  expansion_factor: float = 1.1
94
  mask_json_paths: List[str] = field(default_factory=list) # Now a list
95
 
@@ -105,12 +104,3 @@ class JunctionConfig(BaseConfig):
105
  'thickness': 2,
106
  'font_scale': 0.6
107
  }
108
- =======
109
-
110
- # @dataclass
111
- # class JunctionConfig:
112
- # radius: int = 5
113
- # angle_threshold: float = 25.0
114
- # colinear_threshold: float = 5.0
115
- # connection_threshold: float = 5.0
116
- >>>>>>> temp/test-integration
 
89
  radius: int = 5
90
  angle_threshold_lb: float = 15.0
91
  angle_threshold_ub: float = 75.0
 
92
  expansion_factor: float = 1.1
93
  mask_json_paths: List[str] = field(default_factory=list) # Now a list
94
 
 
104
  'thickness': 2,
105
  'font_scale': 0.6
106
  }
 
 
 
 
 
 
 
 
 
gradioChatApp.py CHANGED
@@ -527,36 +527,25 @@ custom_css = """
527
 
528
  def check_environment():
529
  """Check required environment variables and model files."""
530
- required_vars = {
531
- 'OPENAI_API_KEY': 'OpenAI API key is required for chat functionality',
532
- 'STORAGE_TYPE': 'Storage type must be specified (local or azure)',
533
- 'USE_TORCH': 'Torch configuration must be specified'
534
- }
535
-
536
- missing_vars = []
537
- for var, message in required_vars.items():
538
- if not os.getenv(var):
539
- missing_vars.append(f"{var}: {message}")
540
-
541
- if missing_vars:
542
- logger.error("Missing required environment variables:")
543
- for msg in missing_vars:
544
- logger.error(f" - {msg}")
545
- return False
546
 
547
- # Check for DeepLSD
548
  try:
549
- import deeplsd
550
- logger.info("DeepLSD package found")
551
- except ImportError:
552
- logger.error("DeepLSD package not found. Please install with: pip install git+https://github.com/cvg/DeepLSD.git")
 
 
553
  return False
554
 
555
- # Check for model files
556
- model_path = "models/deeplsd_md.tar"
557
- if not os.path.exists(model_path):
558
- logger.error(f"Required model file not found: {model_path}")
559
- return False
 
 
 
560
 
561
  return True
562
 
@@ -720,12 +709,15 @@ if __name__ == "__main__":
720
  else:
721
  # For Spaces deployment
722
  try:
 
723
  demo = create_ui()
724
  app = demo.app
 
725
  except Exception as e:
726
  logger.error(f"Failed to initialize app: {str(e)}")
727
- # Create minimal error app
728
  with gr.Blocks() as error_demo:
729
  gr.Markdown("# ⚠️ Deployment Error")
730
- gr.Markdown("Failed to initialize the application.")
 
731
  app = error_demo.app
 
527
 
528
  def check_environment():
529
  """Check required environment variables and model files."""
530
+ logger.info("Checking environment configuration...")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
531
 
 
532
  try:
533
+ from storage import StorageFactory
534
+ storage = StorageFactory.get_storage()
535
+ logger.info(f"Storage initialized successfully: {storage.__class__.__name__}")
536
+ except Exception as e:
537
+ logger.error(f"Storage initialization error: {str(e)}")
538
+ logger.error(traceback.format_exc())
539
  return False
540
 
541
+ # Log environment variables (excluding sensitive data)
542
+ env_vars = {
543
+ 'STORAGE_TYPE': os.getenv('STORAGE_TYPE'),
544
+ 'USE_TORCH': os.getenv('USE_TORCH'),
545
+ 'LANGCHAIN_TRACING_V2': os.getenv('LANGCHAIN_TRACING_V2'),
546
+ 'LANGCHAIN_PROJECT': os.getenv('LANGCHAIN_PROJECT')
547
+ }
548
+ logger.info(f"Environment variables: {env_vars}")
549
 
550
  return True
551
 
 
709
  else:
710
  # For Spaces deployment
711
  try:
712
+ logger.info("Initializing Spaces deployment...")
713
  demo = create_ui()
714
  app = demo.app
715
+ logger.info("Application initialized successfully")
716
  except Exception as e:
717
  logger.error(f"Failed to initialize app: {str(e)}")
718
+ logger.error(traceback.format_exc())
719
  with gr.Blocks() as error_demo:
720
  gr.Markdown("# ⚠️ Deployment Error")
721
+ gr.Markdown(f"Error: {str(e)}")
722
+ gr.Markdown("Please check the logs for details.")
723
  app = error_demo.app