Update app.py

app.py

@@ -40,7 +40,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-@tool
+@@tool
 def run_blast(arguments_string: str) -> str:
     """Submits a BLAST job to NCBI and retrieves the results.
 
@@ -162,9 +162,8 @@ def run_blast(arguments_string: str) -> str:
     return result_response.text
 
 if __name__ == "__main__":
-    import os
-    if os.path.basename(sys.argv[0]) == os.path.basename(__file__) and len(sys.argv) > 1:
-        arguments_string = " ".join(sys.argv[1:]) # Reconstruct arguments string from command line
+    if len(sys.argv) > 3: # Expecting at least program, database, and query
+        arguments_string = " ".join(sys.argv[1:])
         results = run_blast(arguments_string)
 
         if results.startswith("Error:"):
@@ -182,7 +181,7 @@ if __name__ == "__main__":
         else:
             print(results)
             sys.exit(0)
-    elif os.path.basename(sys.argv[0]) == os.path.basename(__file__):
+    else: # Run directly but without enough arguments (or no arguments)
         print("This script is intended to be used as a tool function.", file=sys.stderr)
         print("When run directly, it requires command-line arguments:", file=sys.stderr)
         print("Usage: python blast_tool.py program database query_sequence", file=sys.stderr)