Add 'trial mode' - No DB or credits needed

Author: derjogi

app/api/v1/routes/ideas.py

@@ -25,7 +25,7 @@ async def rank_ideas(
     request: Request,
     ideaRequest: IdeaRequest,
     user_info: dict = Depends(verify_token),
-) -> AnalysisResponse:
+) -> AnalysisResponse | Response:
     """
     Analyze and rank ideas based on semantic similarity.
     
@@ -186,17 +186,17 @@ async def process_advanced_features(
     total_bytes: int
 ) -> dict:
     """Process and add advanced features if credits are available"""
-    if request.advanced_features.relationship_graph:
+    if request.advanced_features and request.advanced_features.relationship_graph:
         response["relationship_graph"] = build_relationship_graph(
             response["ranked_ideas"], plot_data
         )
         await CreditService.deduct_credits(user_id, "relationship_graph", num_ideas, total_bytes)
 
-    if request.advanced_features.cluster_names:
+    if request.advanced_features and request.advanced_features.cluster_names:
         response["cluster_names"] = await summarize_clusters(response["ranked_ideas"])
         await CreditService.deduct_credits(user_id, "cluster_names", num_ideas, total_bytes)
 
-    if request.advanced_features.pairwise_similarity_matrix:
+    if request.advanced_features and request.advanced_features.pairwise_similarity_matrix:
         response["pairwise_similarity_matrix"] = plot_data["pairwise_similarity"]
 
     return response

app/core/config.py

@@ -53,6 +53,8 @@ class Settings(BaseSettings):
     USER_DAILY_CREDITS: int = 100
     USER_MAX_CREDITS: int = 1000
 
+    is_in_trial_mode: bool = True
+    
     # Environment
     ENVIRONMENT: str = "DEV"
 

app/core/security.py

@@ -125,6 +125,12 @@ async def verify_token(request: Request, credentials: Optional[HTTPAuthorization
 
     # Whatever the logic flow, first we need to make sure that there isn't a different user still authorized:
     db.auth.sign_out()
+    if settings.is_in_trial_mode:
+        return {
+              "user_id": f"trial_mode_{datetime.now()}", 
+              "is_guest": True, 
+              "balance": 1000
+            }
 
     # Now we can do the rest of the logic. run test routines first, then check guest or proper user.           
     try:

app/services/credits.py

@@ -26,12 +26,16 @@ async def get_total_cost(operations, num_ideas, bytes):
     @staticmethod
     async def get_credits(user_id: str) -> int:
         """Get available credits for user"""
+        if settings.is_in_trial_mode:
+            return 1000
         credits = db.table('credits').select('balance').eq('user_id', user_id).single().execute()
         return int(credits.data['balance']) if credits.data else 0
 
     @staticmethod 
     async def deduct_credits(user_id: str, operation: str, data_size: int, bytes: int) -> bool:
         """Deduct credits after successful operation"""
+        if settings.is_in_trial_mode:
+            return True
         cost = await CreditService.get_operation_cost(operation, data_size, bytes)
         print(f"Deducting {cost} credits from {user_id} for {operation} with {data_size} statements and a total of {bytes} bytes")
         result = db.rpc(