Tweak logging

Author: cyclotruc

src/gitingest/utils.py

@@ -17,6 +17,6 @@ async def wrapper(*args, **kwargs) -> T:
             try:
                 return await asyncio.wait_for(func(*args, **kwargs), timeout=seconds)
             except asyncio.TimeoutError:
-                raise AsyncTimeoutError(f"Operation timed out after {seconds} seconds")
+                raise AsyncTimeoutError(f"Clone timed out after {seconds} seconds")
         return wrapper
     return decorator

src/process_query.py

@@ -11,21 +11,21 @@
 def print_query(query, request, max_file_size, pattern_type, pattern):
     print(f"{Colors.WHITE}{query['url']:<20}{Colors.END}", end="")
     if int(max_file_size/1024) != 50:
-        print(f" | {Colors.PURPLE}Size: {int(max_file_size/1024)}kb{Colors.END}", end="")
+        print(f" | {Colors.YELLOW}Size: {int(max_file_size/1024)}kb{Colors.END}", end="")
     if pattern_type == "include" and pattern != "":
-        print(f" | {Colors.BLUE}Include {pattern}{Colors.END}", end="")
+        print(f" | {Colors.YELLOW}Include {pattern}{Colors.END}", end="")
     elif pattern_type == "exclude" and pattern != "":
         print(f" | {Colors.YELLOW}Exclude {pattern}{Colors.END}", end="")
 
 
 def print_error(query, request, e, max_file_size, pattern_type, pattern):
-    print(f"{Colors.BROWN}WARNING{Colors.END}: {Colors.RED}<- {Colors.END}", end="")
+    print(f"{Colors.BROWN}WARN{Colors.END}: {Colors.RED}<-  {Colors.END}", end="")
     print_query(query, request, max_file_size, pattern_type, pattern)
     print(f" | {Colors.RED}{e}{Colors.END}")
 
 def print_success(query, request, max_file_size, pattern_type, pattern, summary):
     estimated_tokens = summary[summary.index("Estimated tokens:") + len("Estimated ") :]
-    print(f"{Colors.GREEN}INFO{Colors.END}: {Colors.GREEN}<- {Colors.END}", end="")
+    print(f"{Colors.GREEN}INFO{Colors.END}: {Colors.GREEN}<-  {Colors.END}", end="")
     print_query(query, request, max_file_size, pattern_type, pattern)
     print(f" | {Colors.PURPLE}{estimated_tokens}{Colors.END}")
 
@@ -50,10 +50,12 @@ async def process_query(request: Request, input_text: str, slider_position: int,
 
 
     except Exception as e:
+        #hack to print error message when query is not defined
         if 'query' in locals() and query is not None and isinstance(query, dict):
                 print_error(query, request, e, max_file_size, pattern_type, pattern)
         else:
-            print(f"{Colors.RED}Error: {e}{Colors.END}")
+            print(f"{Colors.BROWN}WARN{Colors.END}: {Colors.RED}<-  {Colors.END}", end="")
+            print(f"{Colors.RED}{e}{Colors.END}")
         return templates.TemplateResponse(
             template, 
             {