jebin2 commited on
Commit
3c56e03
·
1 Parent(s): bcc8074
services/gemini_service/__init__.py ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Gemini Service - AI-powered image and video generation
3
+
4
+ Provides:
5
+ - Text generation
6
+ - Image editing
7
+ - Video generation
8
+ - Job processing and background workers
9
+ """
10
+
11
+ # API Client exports
12
+ from services.gemini_service.api_client import (
13
+ GeminiService,
14
+ MODELS,
15
+ DOWNLOADS_DIR,
16
+ get_gemini_api_key,
17
+ MOCK_MODE,
18
+ MOCK_VIDEO_URL,
19
+ )
20
+
21
+ # Job Processor exports
22
+ from services.gemini_service.job_processor import (
23
+ GeminiJobProcessor,
24
+ PriorityWorkerPool,
25
+ get_pool,
26
+ get_priority_for_job_type,
27
+ start_worker,
28
+ stop_worker,
29
+ )
30
+
31
+ __all__ = [
32
+ # API Client
33
+ 'GeminiService',
34
+ 'MODELS',
35
+ 'DOWNLOADS_DIR',
36
+ 'get_gemini_api_key',
37
+ 'MOCK_MODE',
38
+ 'MOCK_VIDEO_URL',
39
+
40
+ # Job Processor
41
+ 'GeminiJobProcessor',
42
+ 'PriorityWorkerPool',
43
+ 'get_pool',
44
+ 'get_priority_for_job_type',
45
+ 'start_worker',
46
+ 'stop_worker',
47
+ ]
services/{gemini_service.py → gemini_service/api_client.py} RENAMED
File without changes
services/{gemini_job_worker.py → gemini_service/job_processor.py} RENAMED
@@ -17,7 +17,7 @@ from services.priority_worker_pool import (
17
  WorkerConfig,
18
  get_interval_for_priority
19
  )
20
- from services.gemini_service import GeminiService
21
  from services.drive_service import DriveService
22
  import asyncio
23
 
 
17
  WorkerConfig,
18
  get_interval_for_priority
19
  )
20
+ from services.gemini_service.api_client import GeminiService
21
  from services.drive_service import DriveService
22
  import asyncio
23